Commit 16bd0d58 authored by haozhenWu's avatar haozhenWu
Browse files

missing import

parent 0f4c1da7

deepchem/models/xgboost_models/__init__.py

+1 −0
Original line number Diff line number Diff line
@@ -10,6 +10,7 @@ from deepchem.utils.save import load_from_disk 
from deepchem.utils.save import save_to_disk

from sklearn.cross_validation import train_test_split

from sklearn.grid_search import GridSearchCV

import tempfile



class XGBoostModel(SklearnModel):

  """

CRA Git | Maintained and supported by SUSTech CRA and CCSE