🐛 fix comment

nissy-dev · Oct 16, 2020 · d104fd7 · d104fd7
1 parent 9a74887
commit d104fd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepchem/models/tests/test_gbdt_model.py b/deepchem/models/tests/test_gbdt_model.py
@@ -128,7 +128,7 @@ def test_classification():
   scores = model.evaluate(test_dataset, [classification_metric])
   assert scores[classification_metric.name] > .9
 
-  # xgboost test
+  # lightgbm test
   lgbm_model = lightgbm.LGBMClassifier(n_estimators=50, seed=123, silent=True)
   model = dc.models.GBDTModel(lgbm_model, **params)
   # fit trained model