def test_store__fetch__remove(self): d = Dataset().data self.db.store(d) self.assertEqual(d, self.db.fetch(d.id, lazy=False)) # self.assertEqual(d, self.db.fetch(d.id, lazy=True)) self.assertEqual(d, self.db.fetch(d)) self.db.removedata(d) self.assertEqual(None, self.db.fetch(d))
def test__rus_(self): augmented_iris = Root >> ( Dataset() * ROS_(strategy={ "virginica": 100, "versicolor": 50, "setosa": 50 })) self.assertEqual(150, len((augmented_iris >> RUS_).X))
def test__rus(self): augmented_iris = Root >> ( Dataset() * ROS_(strategy={ "virginica": 100, "versicolor": 50, "setosa": 50 })) self.assertEqual(200, len( (augmented_iris >> AutoIns * RUS).X)) # TODO: inhibit UserWarning self.assertEqual(150, len((augmented_iris >> AutoIns * RUS).inner.X))
def setUp(self): warnings.simplefilter('ignore', (DeprecationWarning, UserWarning, ImportWarning)) self.db = Tatu("sqlite://:memory:", threaded=False) d = Dataset().data d["X"] = [[1, 2, 3]] d["y"] = [1, 2, 3] self.d = d self.db.store(d) self.letstep = Let(field="X", value=[[1, 2, 3]]) # * Let(field="y", value=[1, 2, 3])
def test__ros(self): iris = Root >> (Dataset() * AutoIns) truncated_iris = iris >> Slice(last=119) rebalanced_iris = truncated_iris >> ROS self.assertEqual( 50, DataFrame(rebalanced_iris.inner.y).value_counts()["virginica"]) larger_iris = iris >> ROS(strategy={ "virginica": 100, "versicolor": 50, "setosa": 50 }) self.assertEqual(200, len(larger_iris.inner.X))
def test__process_(self): truncated_iris = Root >> (Dataset() * Slice(code=":120")) self.assertEqual( 20, DataFrame(truncated_iris.y).value_counts()["virginica"])
def test__process_(self): d = Dataset().data self.assertEqual("19b2d27779bc2d2444c11f5cc24c98ee", mathash("X", d.X)) self.assertEqual("8baa54c6c205d73f99bc1215b7d46c9c", mathash("y", d.y))
def test__rus_(self): samplediris = Root >> (Dataset() * Sample_(n=120)) self.assertEqual(120, len(samplediris.X))
def test__rus(self): samplediris = Root >> (Dataset() * Sample_(n=140)) self.assertEqual(140, len( (samplediris >> AutoIns * Sample).X)) # TODO: inhibit UserWarning self.assertEqual(100, len((samplediris >> AutoIns * Sample).inner.X))