- package com.haolidong.Decisiontree;
- import java.util.ArrayList;
- /**
- *
- * @author haolidong
- * @Description: [该类主要用于保存特征信息以及标签值]
- * @parameter labels: [主要保存标签值]
- */
- public class CreateDataSet extends Matrix {
- public ArrayList < String > labels;
- public CreateDataSet() {
- // TODO Auto-generated constructor stub
- super();
- labels = new ArrayList < String > ();
- }
- /**
- * @author haolidong
- * @Description: [机器学习实战决策树第一个案例的数据]
- */
- public void initTest() {
- ArrayList < String > ab1 = new ArrayList < String > ();
- ArrayList < String > ab2 = new ArrayList < String > ();
- ArrayList < String > ab3 = new ArrayList < String > ();
- ArrayList < String > ab4 = new ArrayList < String > ();
- ArrayList < String > ab5 = new ArrayList < String > ();
- ab1.add("1");
- ab1.add("1");
- ab1.add("yes");
- ab2.add("1");
- ab2.add("1");
- ab2.add("yes");
- ab3.add("1");
- ab3.add("0");
- ab3.add("no");
- ab4.add("0");
- ab4.add("1");
- ab4.add("no");
- ab5.add("0");
- ab5.add("1");
- ab5.add("no");
- data.add(ab1);
- data.add(ab2);
- data.add(ab3);
- data.add(ab4);
- data.add(ab5);
- labels.add("no surfacing");
- labels.add("flippers");
- }
- }
来源: http://lib.csdn.net/snippet/machinelearning/49566