Nevidu commited on
Commit
a045f2c
·
verified ·
1 Parent(s): b3b5d60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -31,6 +31,8 @@ def predict(team, inning, venue, hits, errors, lob, runs, opp_team, opp_runs, op
31
  home_away_status = {'Home': 0, 'Away': 1}
32
  df['Home/Away'] = df['Home/Away'].map(home_away_status)
33
 
 
 
34
  df = pca.transform(df)
35
 
36
  score = xgb_model.predict(df)
@@ -63,6 +65,8 @@ def predict_2(team, inning, venue, hits, errors, lob, runs, opp_team, opp_runs,
63
  home_away_status = {'Home': 0, 'Away': 1}
64
  df['Home/Away'] = df['Home/Away'].map(home_away_status)
65
 
 
 
66
  with open('pca_model1.pkl', 'rb') as f:
67
  pca = pickle.load(f)
68
 
 
31
  home_away_status = {'Home': 0, 'Away': 1}
32
  df['Home/Away'] = df['Home/Away'].map(home_away_status)
33
 
34
+ df = df.astype(int)
35
+
36
  df = pca.transform(df)
37
 
38
  score = xgb_model.predict(df)
 
65
  home_away_status = {'Home': 0, 'Away': 1}
66
  df['Home/Away'] = df['Home/Away'].map(home_away_status)
67
 
68
+ df = df.astype(int)
69
+
70
  with open('pca_model1.pkl', 'rb') as f:
71
  pca = pickle.load(f)
72