patch
parent
4d2a32e75d
commit
3799786c37
|
@ -192,7 +192,7 @@ class eNoseTrainer:
|
||||||
return tmse, mse, mae, rmse, optimized_model, model_params
|
return tmse, mse, mae, rmse, optimized_model, model_params
|
||||||
|
|
||||||
def fit(self):
|
def fit(self):
|
||||||
total_train_queue = self.splits*len(self.get_model_train())
|
total_train_queue = int(1/self.ratio)*len(self.get_model_train())
|
||||||
self.logger.info("{:=^60}".format(f'Begin Fit {total_train_queue} Models'))
|
self.logger.info("{:=^60}".format(f'Begin Fit {total_train_queue} Models'))
|
||||||
self.trained = 0
|
self.trained = 0
|
||||||
manager = enlighten.get_manager()
|
manager = enlighten.get_manager()
|
||||||
|
|
Loading…
Reference in New Issue