Jimin Park commited on
Commit
0605202
·
1 Parent(s): 19e44c2

kermitting soon

Browse files
Files changed (1) hide show
  1. util/app.py +2 -7
util/app.py CHANGED
@@ -209,15 +209,10 @@ def predict_champion(player_opgg_url, *champions):
209
  try:
210
  if model is None:
211
  return "Model not loaded properly"
212
-
213
- print("============= Inside predict_champion(): Model loaded properly=================\n")
214
 
215
  training_df = get_user_training_df(player_opgg_url)
216
 
217
- print("============= Inside predict_champion(): =================\n")
218
  print("training_df type: ", type(training_df), "\n training_df: \n", training_df , "\n")
219
-
220
- print("=============== Inside predict_champion =================== \n")
221
 
222
  training_df = convert_df(training_df)
223
  training_df = apply_feature_engineering(training_df)
@@ -228,8 +223,8 @@ def predict_champion(player_opgg_url, *champions):
228
  target_column='champion',
229
  stratify_columns=['champion', 'region'],
230
  min_samples_per_class=5,
231
- train_size=0,
232
- val_size=1,
233
  random_state=42
234
  )
235
  print("type(X_test): ", type(X_test), "\n")
 
209
  try:
210
  if model is None:
211
  return "Model not loaded properly"
 
 
212
 
213
  training_df = get_user_training_df(player_opgg_url)
214
 
 
215
  print("training_df type: ", type(training_df), "\n training_df: \n", training_df , "\n")
 
 
216
 
217
  training_df = convert_df(training_df)
218
  training_df = apply_feature_engineering(training_df)
 
223
  target_column='champion',
224
  stratify_columns=['champion', 'region'],
225
  min_samples_per_class=5,
226
+ train_size=0.6,
227
+ val_size=0.2,
228
  random_state=42
229
  )
230
  print("type(X_test): ", type(X_test), "\n")