Spaces:
Running
Running
| import gradio as gr | |
| import joblib | |
| import pickle | |
| import numpy as np | |
| 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 | |
| with open('./drive/MyDrive/lgb-NBA.pkl', 'rb') as f: | |
| model = pickle.load(f) | |
| 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="選手情報から年俸を予測(2021-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() |