Spaces:
Runtime error
Runtime error
Commit
·
dfddd70
1
Parent(s):
dbc0b49
debug
Browse files
app.py
CHANGED
@@ -150,11 +150,14 @@ def xgboostPred(df):
|
|
150 |
df = df.drop(['area', 'streetName', 'brf', 'agency'], axis=1)
|
151 |
|
152 |
# Save first row as a numpy array
|
153 |
-
input_list = df.iloc[0].to_numpy()
|
154 |
|
155 |
-
|
|
|
|
|
|
|
|
|
156 |
|
157 |
-
return
|
158 |
|
159 |
def autoPred():
|
160 |
pass
|
|
|
150 |
df = df.drop(['area', 'streetName', 'brf', 'agency'], axis=1)
|
151 |
|
152 |
# Save first row as a numpy array
|
|
|
153 |
|
154 |
+
results = []
|
155 |
+
for _,row in df.iterrows():
|
156 |
+
input_list = row.to_numpy()
|
157 |
+
res = model.predict(np.asarray(input_list).reshape(1, -1))
|
158 |
+
results.append(res[0]) # This is not done in a good way
|
159 |
|
160 |
+
return results
|
161 |
|
162 |
def autoPred():
|
163 |
pass
|