diff --git a/service/tasks.py b/service/tasks.py index a2b088d..9e5a9b1 100644 --- a/service/tasks.py +++ b/service/tasks.py @@ -87,6 +87,7 @@ def raschet(table_name='service_placementpoint', need_time=True, task_name=STATU # Записи для обучения pts_trn = pts.loc[pts.sample_trn == True].reset_index(drop=True) + pts_trn = pts_trn.sort_values('address').reset_index(drop=True) pts_trn = gpd.GeoDataFrame(pts_trn, geometry='geometry', crs='epsg:32637') pts_target = pts_trn[['geometry']] pts_target['cnt'] = 1 @@ -163,7 +164,7 @@ def raschet(table_name='service_placementpoint', need_time=True, task_name=STATU pts_inf['age_day'] = 240 X_inf = pts_inf[feats] - seeds = [3, 99, 87, 21, 15] + seeds = [99, 87, 21, 15] # Обучение, инференс r2_scores = [] @@ -187,7 +188,7 @@ def raschet(table_name='service_placementpoint', need_time=True, task_name=STATU y_infers.append(model.predict(X_inf.drop(columns=['id']))) change_status('Обучение inference 100%', task_name=task_name) # status.status = 'Обучение inference 100%' - current_pred = sum(y_infers) / 5 + current_pred = sum(y_infers) / 4 # расчет шапов explainer = shap.TreeExplainer(model)