Spaces:
Sleeping
Sleeping
taichi motegi
commited on
Commit
•
d9322ec
1
Parent(s):
768991a
Upload 2 files
Browse files- app.py +63 -0
- lgb-NBA.pkl +3 -0
app.py
ADDED
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
import joblib
|
3 |
+
import pickle
|
4 |
+
import numpy as np
|
5 |
+
|
6 |
+
def stats(age, g, gs, mp, fga, tpa, spp, efgp, ft, ftp, orb, drb, ast, stl, blk, tov, pf, pts):
|
7 |
+
stats = []
|
8 |
+
median = 2926620.0
|
9 |
+
dolYen = 130
|
10 |
+
with open('./drive/MyDrive/lgb-NBA.pkl', 'rb') as f:
|
11 |
+
model = pickle.load(f)
|
12 |
+
|
13 |
+
stats.append(age)
|
14 |
+
stats.append(g)
|
15 |
+
stats.append(gs)
|
16 |
+
stats.append(mp)
|
17 |
+
stats.append(fga)
|
18 |
+
stats.append(tpa)
|
19 |
+
stats.append(spp)
|
20 |
+
stats.append(efgp)
|
21 |
+
stats.append(ft)
|
22 |
+
stats.append(ftp)
|
23 |
+
stats.append(orb)
|
24 |
+
stats.append(drb)
|
25 |
+
stats.append(ast)
|
26 |
+
stats.append(stl)
|
27 |
+
stats.append(blk)
|
28 |
+
stats.append(tov)
|
29 |
+
stats.append(pf)
|
30 |
+
stats.append(pts)
|
31 |
+
|
32 |
+
salary = model.predict([stats]) + median
|
33 |
+
salaryUs = round(salary[0])
|
34 |
+
salaryJp = round(salary[0] * dolYen)
|
35 |
+
|
36 |
+
return salaryUs, salaryJp
|
37 |
+
|
38 |
+
demo = gr.Interface(fn=stats,
|
39 |
+
title="選手情報から年俸を予測(2021-NBA)",
|
40 |
+
inputs=[gr.Textbox(label='年齢(Age)'),
|
41 |
+
gr.Textbox(label='試合出場回数(G)'),
|
42 |
+
gr.Textbox(label='スタメン試合出場回数(GS)'),
|
43 |
+
gr.Textbox(label='試合の出場時間(MP)'),
|
44 |
+
gr.Textbox(label='シュート試行回数(FGA)'),
|
45 |
+
gr.Textbox(label='3P試行回数(3PA)'),
|
46 |
+
gr.Textbox(label='2P成功率(2P%)'),
|
47 |
+
gr.Textbox(label='得点期待値(eFG%)'),
|
48 |
+
gr.Textbox(label='フリースロー試行回数(FTA)'),
|
49 |
+
gr.Textbox(label='フリースロー成功率(FT%)'),
|
50 |
+
gr.Textbox(label='オフェンスリバンド(ORB)'),
|
51 |
+
gr.Textbox(label='ディフェンスリバウンド(DRB)'),
|
52 |
+
gr.Textbox(label='アシスト(AST)'),
|
53 |
+
gr.Textbox(label='スティール(STL)'),
|
54 |
+
gr.Textbox(label='ブロック(BLK)'),
|
55 |
+
gr.Textbox(label='ターンオーバー(TOV)'),
|
56 |
+
gr.Textbox(label='ファール(PF)'),
|
57 |
+
gr.Textbox(label='得点(PTS)')
|
58 |
+
],
|
59 |
+
outputs=[gr.Textbox(label='予測年俸($)'),
|
60 |
+
gr.Textbox(label='予測年俸(円)')
|
61 |
+
])
|
62 |
+
|
63 |
+
demo.launch()
|
lgb-NBA.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1c6b768e43aeeaf3c58fdaee6ce65310043c88d97016e74e5f23dfd65eda216d
|
3 |
+
size 21604283
|