Merge pull request #3 from scottwedge/master

Fixes #2
yihong-chen · Feb 22, 2020 · 75cf131 · 75cf131
2 parents c676b71 + ca360b6
commit 75cf131
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/xgb_train_eval_test.py b/xgb_train_eval_test.py
@@ -3,7 +3,7 @@
 import pickle
 from time import time
 from copy import deepcopy
-from importlib imMport reload
+from importlib import reload
 
 import numpy as np
 import pandas as pd
@@ -104,4 +104,4 @@
 with open(constants.XGB_DIR + 'xgb_vip_{}_{:.6f}_{:.6f}'.format(param['booster'],
  test_auc_score,
  res.f1score.mean()), 'wb') as f:
- pickle.dump(bst, f, pickle.HIGHEST_PROTOCOL)
+ pickle.dump(bst, f, pickle.HIGHEST_PROTOCOL)