diff --git a/code/bolsonaro/trainer.py b/code/bolsonaro/trainer.py index c543b853b28dd507002c0ce1bd0cca9048cb4202..aedf7f39b7364bc992fb49a167466c8bca44a0f6 100644 --- a/code/bolsonaro/trainer.py +++ b/code/bolsonaro/trainer.py @@ -166,7 +166,6 @@ class Trainer(object): if type(model) not in [RandomForestRegressor, RandomForestClassifier]: results = ModelRawResults( - model_object='', training_time=self._end_time - self._begin_time, datetime=datetime.datetime.now(), train_score=self.__score_func(model, self._dataset.X_train, self._dataset.y_train, False),