Commit 1bfd15fe authored by Léo Bouscarrat's avatar Léo Bouscarrat
Browse files

Adding the missing argument

parent baa96e2a
......@@ -166,6 +166,7 @@ class Trainer(object):
if type(model) not in [RandomForestRegressor, RandomForestClassifier]:
results = ModelRawResults(
training_time=self._end_time - self._begin_time,,
train_score=self.__score_func(model, self._dataset.X_train, self._dataset.y_train, False),
Supports Markdown
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