def model(titanic_vd):
create_verticapy_schema()
model_class = KNeighborsClassifier("knn_model_test", )
model_class.drop()
model_class.fit("public.titanic", ["age", "fare"], "survived")
yield model_class
model_class.drop()
def test_model_from_vDF(self, titanic_vd):
model_test = KNeighborsClassifier("knn_from_vDF", )
model_test.drop()
model_test.fit(titanic_vd, ["age"], "survived")
assert model_test.score(
cutoff=0.9, method="accuracy") == pytest.approx(0.5890710382513661)
model_test.drop()
def test_contour(self, titanic_vd):
model_test = KNeighborsClassifier("model_contour", )
model_test.drop()
model_test.fit(
titanic_vd,
["age", "fare"],
"survived",
)
result = model_test.contour()
assert len(result.get_default_bbox_extra_artists()) == 34
model_test.drop()
def model(base, titanic_vd):
try:
create_verticapy_schema(base.cursor)
except:
pass
model_class = KNeighborsClassifier("knn_model_test", cursor=base.cursor)
model_class.drop()
model_class.fit("public.titanic", [
"age",
"fare",
], "survived")
yield model_class
model_class.drop()
def test_drop(self):
model_test = KNeighborsClassifier("model_test_drop", )
model_test.drop()
model_test.fit("public.titanic", ["age"], "survived")
current_cursor().execute(
"SELECT model_name FROM verticapy.models WHERE model_name IN ('model_test_drop', '\"model_test_drop\"')"
)
assert current_cursor().fetchone()[0] in (
"model_test_drop",
'"model_test_drop"',
)
model_test.drop()
current_cursor().execute(
"SELECT model_name FROM verticapy.models WHERE model_name IN ('model_test_drop', '\"model_test_drop\"')"
)
assert current_cursor().fetchone() is None
def test_drop(self, base):
model_test = KNeighborsClassifier("model_test_drop",
cursor=base.cursor)
model_test.drop()
model_test.fit("public.titanic", [
"age",
], "survived")
base.cursor.execute(
"SELECT model_name FROM verticapy.models WHERE model_name IN ('model_test_drop', '\"model_test_drop\"')"
)
assert base.cursor.fetchone()[0] in ('model_test_drop',
'"model_test_drop"')
model_test.drop()
base.cursor.execute(
"SELECT model_name FROM verticapy.models WHERE model_name IN ('model_test_drop', '\"model_test_drop\"')"
)
assert base.cursor.fetchone() is None
def test_repr(self, model):
assert "Additional Info" in model.__repr__()
model_repr = KNeighborsClassifier("model_repr")
model_repr.drop()
assert model_repr.__repr__() == "<KNeighborsClassifier>"