Package aima.core.learning.framework

Examples of aima.core.learning.framework.DataSet

49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
  @Test
  public void testClassificationReturnedWhenAllExamplesHaveTheSameClassification()
      throws Exception {
    // tests RecursionBaseCase#2
    DataSet ds = DataSetFactory.getRestaurantDataSet();
    DecisionTreeLearner learner = new DecisionTreeLearner();

    DataSet ds2 = ds.emptyDataSet();

    // all 3 examples have the same classification (willWait = yes)
    ds2.add(ds.getExample(0));
    ds2.add(ds.getExample(2));
    ds2.add(ds.getExample(3));

    learner.train(ds2);
    Assert.assertEquals("Yes", learner.predict(ds.getExample(0)));
  }
View Full Code Here
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
  @Test
  public void testMajorityReturnedWhenAttributesToExamineIsEmpty()
      throws Exception {
    // tests RecursionBaseCase#2
    DataSet ds = DataSetFactory.getRestaurantDataSet();
    DecisionTreeLearner learner = new DecisionTreeLearner();

    DataSet ds2 = ds.emptyDataSet();

    // 3 examples have classification = "yes" and one ,"no"
    ds2.add(ds.getExample(0));
    ds2.add(ds.getExample(1));// "no"
    ds2.add(ds.getExample(2));
    ds2.add(ds.getExample(3));
    ds2.setSpecification(new MockDataSetSpecification("will_wait"));

    learner.train(ds2);
    Assert.assertEquals("Yes", learner.predict(ds.getExample(1)));
  }
View Full Code Here
85
86
87
88
89
90
91
92
93
94
95
    Assert.assertEquals("Yes", learner.predict(ds.getExample(1)));
  }

  @Test
  public void testInducedTreeClassifiesDataSetCorrectly() throws Exception {
    DataSet ds = DataSetFactory.getRestaurantDataSet();
    DecisionTreeLearner learner = new DecisionTreeLearner();
    learner.train(ds);
    int[] result = learner.test(ds);
    Assert.assertEquals(12, result[0]);
    Assert.assertEquals(0, result[1]);
View Full Code Here
99
100
101
102
103
104
105
106
107
108
109
110
  public void testDecisionListLearnerReturnsNegativeDLWhenDataSetEmpty()
      throws Exception {
    // tests first base case of DL Learner
    DecisionListLearner learner = new DecisionListLearner("Yes", "No",
        new MockDLTestFactory(null));
    DataSet ds = DataSetFactory.getRestaurantDataSet();
    DataSet empty = ds.emptyDataSet();
    learner.train(empty);
    Assert.assertEquals("No", learner.predict(ds.getExample(0)));
    Assert.assertEquals("No", learner.predict(ds.getExample(1)));
    Assert.assertEquals("No", learner.predict(ds.getExample(2)));
  }
View Full Code Here
113
114
115
116
117
118
119
120
121
122
  public void testDecisionListLearnerReturnsFailureWhenTestsEmpty()
      throws Exception {
    // tests second base case of DL Learner
    DecisionListLearner learner = new DecisionListLearner("Yes", "No",
        new MockDLTestFactory(new ArrayList<DLTest>()));
    DataSet ds = DataSetFactory.getRestaurantDataSet();
    learner.train(ds);
    Assert.assertEquals(DecisionListLearner.FAILURE,
        learner.predict(ds.getExample(0)));
  }
View Full Code Here
121
122
123
124
125
126
127
128
129
130
131
        learner.predict(ds.getExample(0)));
  }

  @Test
  public void testDecisionListTestRunOnRestaurantDataSet() throws Exception {
    DataSet ds = DataSetFactory.getRestaurantDataSet();
    DecisionListLearner learner = new DecisionListLearner("Yes", "No",
        new DLTestFactory());
    learner.train(ds);

    int[] result = learner.test(ds);
View Full Code Here
133
134
135
136
137
138
139
140
141
142
143
    Assert.assertEquals(0, result[1]);
  }

  @Test
  public void testCurrentBestLearnerOnRestaurantDataSet() throws Exception {
    DataSet ds = DataSetFactory.getRestaurantDataSet();
    CurrentBestLearner learner = new CurrentBestLearner("Yes");
    learner.train(ds);

    int[] result = learner.test(ds);
    Assert.assertEquals(12, result[0]);
View Full Code Here
22
23
24
25
26
27
28
29
30
31
32
  private static final String YES = "Yes";

  @Test
  public void testAdaBoostEnablesCollectionOfStumpsToClassifyDataSetAccurately()
      throws Exception {
    DataSet ds = DataSetFactory.getRestaurantDataSet();
    List<DecisionTree> stumps = DecisionTree.getStumpsFor(ds, YES, "No");
    List<Learner> learners = new ArrayList<Learner>();
    for (Object stump : stumps) {
      DecisionTree sl = (DecisionTree) stump;
      StumpLearner stumpLearner = new StumpLearner(sl, "No");
View Full Code Here
64
65
66
67
68
69
70
71
72
73
74
75
76
77
  }

  @Test
  public void testLoadsDatasetFile() throws Exception {

    DataSet ds = DataSetFactory.getRestaurantDataSet();
    Assert.assertEquals(12, ds.size());

    Example first = ds.getExample(0);
    Assert.assertEquals(YES, first.getAttributeValueAsString("alternate"));
    Assert.assertEquals("$$$", first.getAttributeValueAsString("price"));
    Assert.assertEquals("0-10",
        first.getAttributeValueAsString("wait_estimate"));
    Assert.assertEquals(YES, first.getAttributeValueAsString("will_wait"));
View Full Code Here
84
85
86
87
88
89
90
91
92
93
  }

  @Test
  public void testLoadsIrisDataSetWithNumericAndStringAttributes()
      throws Exception {
    DataSet ds = DataSetFactory.getIrisDataSet();
    Example first = ds.getExample(0);
    Assert.assertEquals("5.1",
        first.getAttributeValueAsString("sepal_length"));
  }
View Full Code Here
TOP

Related Classes of aima.core.learning.framework.DataSet

  • aima.core.learning.inductive.DLTest
  • aima.core.learning.learners.DecisionListLearner
  • aima.core.learning.learners.DecisionTreeLearner
  • aima.gui.demo.learning.LearningDemo
  • aima.test.core.unit.learning.framework.DataSetTest
  • aima.test.core.unit.learning.framework.InformationAndGainTest
  • aima.test.core.unit.learning.inductive.DecisionListTest
  • aima.test.core.unit.learning.inductive.DLTestTest
  • aima.test.core.unit.learning.learners.EnsembleLearningTest
  • aima.test.core.unit.learning.learners.LearnerTests
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.