diff --git a/service/tasks.py b/service/tasks.py index 74f7d68..2571c0b 100644 --- a/service/tasks.py +++ b/service/tasks.py @@ -180,7 +180,7 @@ def raschet(table_name='service_placementpoint', need_time=True, task_name=STATU # status.status = 'Обучение inference: ' + str(int((seeds.index(i) + 1) / len(seeds) * 100)) + '%' # status.save() #change_status(f'Обучение inference: {str(int((seeds.index(i) + 1) / len(seeds) * 100))}%', - task_name=task_name) + #task_name=task_name) x_trn, x_test, y_trn, y_test = ms.train_test_split(X_trn, Y_trn, test_size=0.2, random_state=i) model = catboost.CatBoostRegressor(cat_features=['property_era'], random_state=i) model.fit(x_trn, y_trn, verbose=False)