diff --git a/code/bolsonaro/trainer.py b/code/bolsonaro/trainer.py index aedf7f39b7364bc992fb49a167466c8bca44a0f6..0e437d223ddabdac1e9a020a05e9a3691b6e3d0e 100644 --- a/code/bolsonaro/trainer.py +++ b/code/bolsonaro/trainer.py @@ -166,6 +166,7 @@ class Trainer(object): if type(model) not in [RandomForestRegressor, RandomForestClassifier]: results = ModelRawResults( + model_weights='', 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),