コード例 #1
0
ファイル: TestKNN.py プロジェクト: yoyzhou/dmP
class Test(unittest.TestCase):
    def setUp(self):
        self.ds = DataSource("../data.o", False)

        self.knn = KNN(self.ds.dataset[:-2], self.ds.dataset[7], 4)

    def tearDown(self):
        pass

    def testDistanceMetric(self):
        self.assertEqual(self.knn.distanceMetric([0, 4, 6], [0, 1, 2]), 5)

    def testClassfySingle(self):
        self.knn = KNN(self.ds.dataset[:-2], self.ds.dataset[7], 4)
        self.assertEqual(self.knn.classify(True, 3), "will buy")
        # self.knn.classify();
        # print([rtn for rtn in self.knn.queryset]) self.knn = KNN(self.ds.dataset[:-2], self.ds.dataset[7], 4)

    def testClassfyMultiple(self):
        self.knn = KNN(self.ds.dataset[3:], self.ds.dataset[0:3], 4)
        self.knn.classify(True, 3)
        print([rtn for rtn in self.knn.queryset])

    def testName(self):
        pass
コード例 #2
0
ファイル: TestKNN.py プロジェクト: yoyzhou/dmP
 def testClassfyMultiple(self):
     self.knn = KNN(self.ds.dataset[3:], self.ds.dataset[0:3], 4)
     self.knn.classify(True, 3)
     print([rtn for rtn in self.knn.queryset])
コード例 #3
0
ファイル: TestKNN.py プロジェクト: yoyzhou/dmP
 def testClassfySingle(self):
     self.knn = KNN(self.ds.dataset[:-2], self.ds.dataset[7], 4)
     self.assertEqual(self.knn.classify(True, 3), "will buy")
コード例 #4
0
ファイル: TestKNN.py プロジェクト: yoyzhou/dmP
    def setUp(self):
        self.ds = DataSource("../data.o", False)

        self.knn = KNN(self.ds.dataset[:-2], self.ds.dataset[7], 4)