File size: 1,038 Bytes
e405448 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | import pickle
import gradio as gr
import pandas as pd
def rf_predict(GDP, Unemployment, Medical_resources, Past_years_life_expectancy_growth_rate):
# 读取模型
with open('rf.pkl', 'rb') as f:
rf = pickle.load(f)
# 预测
X_test = [GDP, Unemployment, Medical_resources, Past_years_life_expectancy_growth_rate]
X_test = pd.DataFrame(X_test).T
X_test.columns = ['GDP_per_capita', 'Unemployment_rate', 'Medical_resources_per_capita',
'Past_years_life_expectancy_growth_rate']
y_pred = rf.predict(X_test)
return y_pred
# def rf_predict(GDP, Unemployment, Medical_resources, Past_years_life_expectancy_growth_rate):
# print(rf_predict(85698, 5.5, 2120, 0.1))
gr.Interface(
fn=rf_predict,
inputs=[
gr.Textbox(label="GDP"),
gr.Textbox(label="Unemployment"),
gr.Textbox(label="Medical_resources"),
gr.Textbox(label="Past_years_life_expectancy_growth_rate"),
],
outputs="number",
title="Life Expectancy Prediction"
).launch()
|