Nathanotal commited on
Commit
dfddd70
·
1 Parent(s): dbc0b49
Files changed (1) hide show
  1. app.py +6 -3
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
- res = model.predict(np.asarray(input_list).reshape(1, -1))
 
 
 
 
156
 
157
- return res
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