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