hlnicholls commited on
Commit
d5fefc1
1 Parent(s): 037b3d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ xgb = xgboost.XGBRegressor(
34
 
35
  xgb.fit(X, Y)
36
 
37
- prediction_list = list(xgb.predict(data))
38
  predictions = [round(prediction, 2) for prediction in prediction_list]
39
 
40
  output = pd.Series(data=predictions, index=annotations.index, name="XGB_Score")
 
34
 
35
  xgb.fit(X, Y)
36
 
37
+ prediction_list = list(xgb.predict(annotations))
38
  predictions = [round(prediction, 2) for prediction in prediction_list]
39
 
40
  output = pd.Series(data=predictions, index=annotations.index, name="XGB_Score")