Spaces:
Build error
Build error
Update model.py
Browse files
model.py
CHANGED
|
@@ -55,6 +55,6 @@ f1 = f1_score(pred, y_test)
|
|
| 55 |
print(f"F1 Score for LightGBM: {f1}.")
|
| 56 |
|
| 57 |
# Using pickle to save model
|
| 58 |
-
lightgbm = open("
|
| 59 |
pkl.dump(lgbm, lightgbm)
|
| 60 |
lightgbm.close()
|
|
|
|
| 55 |
print(f"F1 Score for LightGBM: {f1}.")
|
| 56 |
|
| 57 |
# Using pickle to save model
|
| 58 |
+
lightgbm = open("./lightgbm.pickle", "wb")
|
| 59 |
pkl.dump(lgbm, lightgbm)
|
| 60 |
lightgbm.close()
|