Update tasks.py

dev
Oleg Bogatkin 2 years ago
parent 7e044ac854
commit 86f1fe165a

@ -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.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_trn = gpd.GeoDataFrame(pts_trn, geometry='geometry', crs='epsg:32637')
pts_target = pts_trn[['geometry']] pts_target = pts_trn[['geometry']]
pts_target['cnt'] = 1 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 pts_inf['age_day'] = 240
X_inf = pts_inf[feats] X_inf = pts_inf[feats]
seeds = [3, 99, 87, 21, 15] seeds = [99, 87, 21, 15]
# Обучение, инференс # Обучение, инференс
r2_scores = [] 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']))) y_infers.append(model.predict(X_inf.drop(columns=['id'])))
change_status('Обучение inference 100%', task_name=task_name) change_status('Обучение inference 100%', task_name=task_name)
# status.status = 'Обучение inference 100%' # status.status = 'Обучение inference 100%'
current_pred = sum(y_infers) / 5 current_pred = sum(y_infers) / 4
# расчет шапов # расчет шапов
explainer = shap.TreeExplainer(model) explainer = shap.TreeExplainer(model)

Loading…
Cancel
Save