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
|