Python KNNの例

コード例 #1

ファイル: 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

ファイル: 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

ファイル: 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

ファイル: 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)