def testSplitDataset_fractions(self):
h = [0, 1, 2, 3]
listdataset = dataset.ListDataset(elem_list=h)
splitdataset = dataset.SplitDataset(listdataset, {
'train': 0.75,
'val': 0.25
})
splitdataset.select('train')
self.assertEqual(len(splitdataset), 3)
self.assertEqual(splitdataset[2], 2)
splitdataset.select('val')
self.assertEqual(len(splitdataset), 1)
self.assertEqual(splitdataset[0], 3)
def testSplitDataset(self):
h = [0, 1, 2, 3]
listdataset = dataset.ListDataset(elem_list=h)
splitdataset = dataset.SplitDataset(listdataset, {
'train': 3,
'val': 1
})
splitdataset.select('train')
self.assertEqual(len(splitdataset), 3)
self.assertEqual(splitdataset[2], 2)
splitdataset.select('val')
self.assertEqual(len(splitdataset), 1)
self.assertEqual(splitdataset[0], 3)
# test fluent api
splitdataset = listdataset.split({'train': 3, 'val': 1})
splitdataset.select('train')
self.assertEqual(len(splitdataset), 3)
self.assertEqual(splitdataset[2], 2)
def testSplitDataset(self):
h = [0, 1, 2, 3]
listdataset = dataset.ListDataset(elem_list=h)
splitdataset = dataset.SplitDataset(listdataset, {
"train": 3,
"val": 1
})
splitdataset.select("train")
self.assertEqual(len(splitdataset), 3)
self.assertEqual(splitdataset[2], 2)
splitdataset.select("val")
self.assertEqual(len(splitdataset), 1)
self.assertEqual(splitdataset[0], 3)
# test fluent api
splitdataset = listdataset.split({"train": 3, "val": 1})
splitdataset.select("train")
self.assertEqual(len(splitdataset), 3)
self.assertEqual(splitdataset[2], 2)
