"python-package/vscode:/vscode.git/clone" did not exist on "7c1e92dd8e5f17362d6de92f5cda36d6a6b8ff79"
Commit cf4edf0e authored by Guolin Ke's avatar Guolin Ke
Browse files

add test for check model file persistence

parent dcbdc675
...@@ -37,7 +37,13 @@ class TestBasic(unittest.TestCase): ...@@ -37,7 +37,13 @@ class TestBasic(unittest.TestCase):
os.remove(tname) os.remove(tname)
self.assertEqual(len(pred_from_matr), len(pred_from_file)) self.assertEqual(len(pred_from_matr), len(pred_from_file))
for preds in zip(pred_from_matr, pred_from_file): for preds in zip(pred_from_matr, pred_from_file):
self.assertAlmostEqual(*preds, places=5) self.assertAlmostEqual(*preds, places=15)
#check saved model persistence
bst = lgb.Booster(params, model_file="model.txt")
pred_from_model_file = bst.predict(X_test)
self.assertEqual(len(pred_from_matr), len(pred_from_model_file))
for preds in zip(pred_from_matr, pred_from_model_file):
self.assertAlmostEqual(*preds, places=15)
print("----------------------------------------------------------------------") print("----------------------------------------------------------------------")
print("running test_basic.py") print("running test_basic.py")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment