Преглед на файлове

generate label for mysql

罗俊辉 преди 1 година
родител
ревизия
4cc3b354e6
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      main_spider.py

+ 1 - 1
main_spider.py

@@ -58,7 +58,7 @@ class LightGBM(object):
         """
         path = "data/train_data/spider_data_240401.json"
         df = pd.read_json(path)
-        df.dropna(subset=['label'])
+        df = df.dropna(subset=['label'])
         labels = df['label']
         features = df.drop("label", axis=1)
         for key in self.float_columns: