FilesExpand file tree

master

/

DataSetLoader.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

52 lines (37 loc) · 1.51 KB

master

/

DataSetLoader.java

Copy path

Top

File metadata and controls

• Code
• Blame

52 lines (37 loc) · 1.51 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

package utils;

import datastructs.maths.DenseMatrixSet;

import datastructs.maths.RowBuilder;

import datastructs.utils.RowType;

import tech.tablesaw.api.Table;

import tech.tablesaw.columns.Column;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

public class DataSetLoader {

public static Pair<DenseMatrixSet<Double>, List<Integer>> loadIrisDataSet() throws IOException, IllegalArgumentException {

// load the data

Table dataSetTable = TableDataSetLoader.loadDataSet(new File("src/main/resources/datasets/iris_data.csv"));

List<Integer> labels = new ArrayList<>();

Column species = dataSetTable.column("species");

for (int i = 0; i < species.size(); i++) {

String label = (String) species.get(i);

if(label.equals("Iris-setosa")){

labels.add(0);

}

else if(label.equals("Iris-versicolor")){

labels.add(1);

}

else if(label.equals("Iris-virginica")){

labels.add(2);

}

else{

throw new IllegalArgumentException("Unknown class");

}

}

Table reducedDataSet = dataSetTable.removeColumns("species").first(dataSetTable.rowCount());

DenseMatrixSet<Double> dataSet = new DenseMatrixSet<Double>(RowType.Type.DOUBLE_VECTOR, new RowBuilder());

dataSet.initializeFrom(reducedDataSet);

return PairBuilder.makePair(dataSet, labels);

}

}