Update tasks.py

2 years ago · 135ebc9037
parent cc3b65ffcc
commit 135ebc9037
1 changed files with 1 additions and 1 deletions
--- a/service/tasks.py
+++ b/service/tasks.py
@ -86,7 +86,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.name == 'постамат Яндекс.Маркет из обучающей выборки'].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']]