taichi motegi commited on
Commit
d9322ec
1 Parent(s): 768991a

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +63 -0
  2. 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