"git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "a178b75b676735a1ec300cf82eba2b31a50cf8bc"
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):
os.remove(tname)
self.assertEqual(len(pred_from_matr), len(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("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