dataset_tests.py 文件源码

def test_training_set(self):
        dataset = ds.Dataset("realDonaldTrump", "HillaryClinton", 0.8)
        # should have 50% positive and 50% negative examples
        trainSet = dataset.trainSet
        validationSet = dataset.validationSet
        tCount = ds.get_positive_negative_amount(trainSet)
        vCount = ds.get_positive_negative_amount(validationSet)
        self.assertEqual(tCount[0], tCount[1])
        self.assertEqual(vCount[0], vCount[1])