Spaces:
Sleeping
Sleeping
import gradio as gr | |
import pickle | |
with open('./lgb-NBA.pkl', 'rb') as f: | |
model = pickle.load(f) | |
def stats(age, g, gs, mp, fga, tpa, spp, efgp, ft, ftp, orb, drb, ast, stl, blk, tov, pf, pts): | |
stats = [] | |
median = 2926620.0 | |
dolYen = 130 | |
stats.append(age) | |
stats.append(g) | |
stats.append(gs) | |
stats.append(mp) | |
stats.append(fga) | |
stats.append(tpa) | |
stats.append(spp) | |
stats.append(efgp) | |
stats.append(ft) | |
stats.append(ftp) | |
stats.append(orb) | |
stats.append(drb) | |
stats.append(ast) | |
stats.append(stl) | |
stats.append(blk) | |
stats.append(tov) | |
stats.append(pf) | |
stats.append(pts) | |
salary = model.predict([stats]) + median | |
salaryUs = round(salary[0]) | |
salaryJp = round(salary[0] * dolYen) | |
return salaryUs, salaryJp | |
demo = gr.Interface(fn=stats, | |
title="選手情報から年俸を予測(NBA)", | |
inputs=[gr.Textbox(label='年齢(Age)'), | |
gr.Textbox(label='試合出場回数(G)'), | |
gr.Textbox(label='スタメン試合出場回数(GS)'), | |
gr.Textbox(label='試合の出場時間(MP)'), | |
gr.Textbox(label='シュート試行回数(FGA)'), | |
gr.Textbox(label='3P試行回数(3PA)'), | |
gr.Textbox(label='2P成功率(2P%)'), | |
gr.Textbox(label='得点期待値(eFG%)'), | |
gr.Textbox(label='フリースロー試行回数(FTA)'), | |
gr.Textbox(label='フリースロー成功率(FT%)'), | |
gr.Textbox(label='オフェンスリバンド(ORB)'), | |
gr.Textbox(label='ディフェンスリバウンド(DRB)'), | |
gr.Textbox(label='アシスト(AST)'), | |
gr.Textbox(label='スティール(STL)'), | |
gr.Textbox(label='ブロック(BLK)'), | |
gr.Textbox(label='ターンオーバー(TOV)'), | |
gr.Textbox(label='ファール(PF)'), | |
gr.Textbox(label='得点(PTS)') | |
], | |
outputs=[gr.Textbox(label='予測年俸($)'), | |
gr.Textbox(label='予測年俸(円)') | |
]) | |
demo.launch() |