Merge branch 'o.bogatkin-dev-patch-58030' into 'dev'

Update tasks.py

See merge request spatial/postamates!182
dev
Oleg Bogatkin 2 years ago
commit 2c08689cdd

@ -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)

Loading…
Cancel
Save