From e03d8b48af6c2810fb0ee4d0473c91606327cf70 Mon Sep 17 00:00:00 2001 From: LukaszChrostowski Date: Mon, 8 Jan 2024 23:02:58 +0100 Subject: [PATCH] work work work --- .DS_Store | Bin 8196 -> 8196 bytes app/src/flask-server/modele/modele.py | 9 ++++++--- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/.DS_Store b/.DS_Store index 4b2dabf0f183b068696d0d561ff32d7cec30e8b7..b06a9fd2706ace93b89e2fd4763d561e9067b5f0 100644 GIT binary patch delta 24 fcmZp1XmQxESAgBjSVzIg++gy35&O-|f{gqCV;~0i delta 16 XcmZp1XmQxES77oD5x33Ef{gqCIdKJ| diff --git a/app/src/flask-server/modele/modele.py b/app/src/flask-server/modele/modele.py index f9d1884..db1c82e 100644 --- a/app/src/flask-server/modele/modele.py +++ b/app/src/flask-server/modele/modele.py @@ -99,7 +99,7 @@ def xgboost_predict_proba_v2(shooter,goalkeeper,teamMatesList,opponentsList, min 'shot_open_goal','shot_follows_dribble', 'shot_redirect', 'x1', 'y1','number_of_players_opponents', 'number_of_players_teammates','angle', 'distance', - 'x_player_opponent_Goalkeeper', + 'x_player_opponent_Goalkeeper', 'x_player_opponent_8', 'x_player_opponent_1', 'x_player_opponent_2','x_player_opponent_3', 'x_player_opponent_4','x_player_opponent_5', 'x_player_opponent_6', 'x_player_teammate_2','x_player_opponent_9', 'x_player_opponent_10', @@ -156,7 +156,7 @@ def xgboost_predict_proba_v2(shooter,goalkeeper,teamMatesList,opponentsList, min shot_open_goal,shot_follows_dribble, shot_redirect, shooter[0], shooter[1],number_of_players_opponents, number_of_players_teammates,angle, distance, - x_player_opponent_Goalkeeper, + x_player_opponent_Goalkeeper, opponentsList[8][0], opponentsList[0][0], opponentsList[1][0], opponentsList[2][0], opponentsList[3][0], opponentsList[4][0], opponentsList[5][0], opponentsList[6][0], opponentsList[7][0], teamMatesList[8][0], @@ -176,7 +176,10 @@ def xgboost_predict_proba_v2(shooter,goalkeeper,teamMatesList,opponentsList, min x_player_teammate_Goalkeeper, y_player_teammate_Goalkeeper, shot_kick_off] - + categorical_columns = ['position_name', 'shot_technique_name', 'shot_type_name', 'number_of_players_opponents', 'number_of_players_teammates', 'shot_body_part_name'] # list all your object columns here + # X_new = pd.get_dummies(X_new, columns=categorical_columns) + X_new[categorical_columns] = X_new[categorical_columns].astype('category') + return model.predict_proba(X_new)[0][1].round(2)