-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathapp.py
98 lines (81 loc) · 4.31 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
from resources.functions import *
def run():
brawlers = pd.read_parquet('datasets/brawlers/brawlers.parquet')
maps = pd.read_parquet('datasets/maps/maplist.parquet')
metrics = pd.read_json('resources/bs_metrics.json', orient='index')
means = pd.read_json('resources/means.json')
means = round(means, 0)
mean_brawler_trophies = int(means['mean_brawler_trophies'][0])
mean_brawler_power = int(means['mean_brawler_power'][0])
models_dict = load_models()
st.sidebar.subheader('Model Metrics')
st.sidebar.dataframe(metrics, use_container_width=True)
st.title('BrawlStars ML Model')
st.write('This is a simple ML model that predicts the outcome of a BrawlStars match based on the brawlers and maps chosen.')
st.write('The model is trained on data from the BrawlStars API, and the data is available in the sidebar.')
buy_me_a_coffee(username="aquinteros", floating=False, width=221)
ct = st.container()
with ct:
st.header('Game Settings')
c1, c2 = st.columns(2)
event_mode = c1.selectbox('Game Mode', maps['event_mode'].unique())
event_map = c2.selectbox('Map', maps[maps['event_mode'] == event_mode]['event_map'])
st.header('Team 1')
c1, c2, c3 = st.columns(3)
c1.subheader('Player 1')
t1p1_name = c1.selectbox('Brawler', brawlers['name'], key='t1p1_name')
t1p1_power = c1.slider('Power', 1, 11, mean_brawler_power, key='t1p1_power')
t1p1_brawler_trophies = c1.slider('Brawler Trophies', 0, 1_500, mean_brawler_trophies, key='t1p1_brawler_trophies')
c2.subheader('Player 2')
t1p2_name = c2.selectbox('Brawler', brawlers['name'], key='t1p2_name')
t1p2_power = c2.slider('Power', 1, 11, mean_brawler_power, key='t1p2_power')
t1p2_brawler_trophies = c2.slider('Brawler Trophies', 0, 1_500, mean_brawler_trophies, key='t1p2_brawler_trophies')
c3.subheader('Player 3')
t1p3_name = c3.selectbox('Brawler', brawlers['name'], key='t1p3_name')
t1p3_power = c3.slider('Power', 1, 11, mean_brawler_power, key='t1p3_power')
t1p3_brawler_trophies = c3.slider('Brawler Trophies', 0, 1_500, mean_brawler_trophies, key='t1p3_brawler_trophies')
st.header('Team 2')
c1, c2, c3 = st.columns(3)
c1.subheader('Player 1')
t2p1_name = c1.selectbox('Brawler', brawlers['name'], key='t2p1_name')
t2p1_power = c1.slider('Power', 1, 11, mean_brawler_power, key='t2p1_power')
t2p1_brawler_trophies = c1.slider('Brawler Trophies', 0, 1_500, mean_brawler_trophies, key='t2p1_brawler_trophies')
c2.subheader('Player 2')
t2p2_name = c2.selectbox('Brawler', brawlers['name'], key='t2p2_name')
t2p2_power = c2.slider('Power', 1, 11, mean_brawler_power, key='t2p2_power')
t2p2_brawler_trophies = c2.slider('Brawler Trophies', 0, 1_500, mean_brawler_trophies, key='t2p2_brawler_trophies')
c3.subheader('Player 3')
t2p3_name = c3.selectbox('Brawler', brawlers['name'], key='t2p3_name')
t2p3_power = c3.slider('Power', 1, 11, mean_brawler_power, key='t2p3_power')
t2p3_brawler_trophies = c3.slider('Brawler Trophies', 0, 1_500, mean_brawler_trophies, key='t2p3_brawler_trophies')
input_df = {
'event_mode': event_mode,
'event_map': event_map,
'battle_team1_player1_brawler_name': t1p1_name,
'battle_team1_player2_brawler_name': t1p2_name,
'battle_team1_player3_brawler_name': t1p3_name,
'battle_team2_player1_brawler_name': t2p1_name,
'battle_team2_player2_brawler_name': t2p2_name,
'battle_team2_player3_brawler_name': t2p3_name,
'battle_team1_player1_brawler_power': t1p1_power,
'battle_team1_player2_brawler_power': t1p2_power,
'battle_team1_player3_brawler_power': t1p3_power,
'battle_team2_player1_brawler_power': t2p1_power,
'battle_team2_player2_brawler_power': t2p2_power,
'battle_team2_player3_brawler_power': t2p3_power,
'battle_team1_player1_brawler_trophies': t1p1_brawler_trophies,
'battle_team1_player2_brawler_trophies': t1p2_brawler_trophies,
'battle_team1_player3_brawler_trophies': t1p3_brawler_trophies,
'battle_team2_player1_brawler_trophies': t2p1_brawler_trophies,
'battle_team2_player2_brawler_trophies': t2p2_brawler_trophies,
'battle_team2_player3_brawler_trophies': t2p3_brawler_trophies,
}
if st.button('Predict'):
output = predict(input_df, models_dict[event_mode])
st.markdown(f"""
# Prediction
The winner is the team number **{output['prediction_label'].values}**
The probability of winning is **{output['prediction_score'].values}**
""")
if __name__ == '__main__':
run()