def test_save_to_Path():
model = Workspace(quiet=True)
model.learn("1 | a b c")
model.save(Path("tmp1.model"))
model.save("tmp2.model")
assert filecmp.cmp("tmp1.model", "tmp2.model")
os.remove("tmp1.model")
os.remove("tmp2.model")
def test_regressor_args():
# load and parse external data file
data_file = os.path.join(os.path.dirname(os.path.realpath(__file__)),
"resources", "train.dat")
model = Workspace(oaa=3, data=data_file, passes=30, c=True, k=True)
assert model.predict("| feature1:2.5") == 1
# update model in memory
for _ in range(10):
model.learn("3 | feature1:2.5")
assert model.predict("| feature1:2.5") == 3
# save model
model.save("tmp.model")
del model
# load initial regressor and confirm updated prediction
new_model = Workspace(i="tmp.model", quiet=True)
assert new_model.predict("| feature1:2.5") == 3
del new_model
# clean up
os.remove("{}.cache".format(data_file))
os.remove("tmp.model")