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