Spaces:

ivwhy
/

lol_champion_pick_predictor

Sleeping

Jimin Park commited on Jan 13

Commit

6af173b

1 Parent(s): 5e39132

kermitting soon

Files changed (1) hide show

util/app.py CHANGED Viewed

@@ -167,7 +167,7 @@ def predict_champion(player_opgg_url, *champions):
         X_processed = X_processed.astype('float32')
         # Create DMatrix and predict
-        dtest = DMatrix(X_processed, enable_categorical=True)
         predictions = model.predict(dtest)
         # Get prediction indices

         X_processed = X_processed.astype('float32')
         # Create DMatrix and predict
+        dtest = DMatrix(X_processed, enable_categorical=False)
         predictions = model.predict(dtest)
         # Get prediction indices