InEase/Fintech

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