Spaces:
Runtime error
Runtime error
| import joblib | |
| import pandas as pd | |
| EDU_DICT = {'Preschool': 1, | |
| '1st-4th': 2, | |
| '5th-6th': 3, | |
| '7th-8th': 4, | |
| '9th': 5, | |
| '10th': 6, | |
| '11th': 7, | |
| '12th': 8, | |
| 'HS-grad': 9, | |
| 'Some-college': 10, | |
| 'Assoc-voc': 11, | |
| 'Assoc-acdm': 12, | |
| 'Bachelors': 13, | |
| 'Masters': 14, | |
| 'Prof-school': 15, | |
| 'Doctorate': 16 | |
| } | |
| model = # | |
| unique_values = # | |
| unique_class = unique_values["workclass"] | |
| unique_education = unique_values["education"] | |
| unique_marital_status = unique_values["marital.status"] | |
| unique_relationship = unique_values["relationship"] | |
| unique_occupation = unique_values["occupation"] | |
| unique_sex = unique_values["sex"] | |
| unique_race = unique_values["race"] | |
| unique_country = unique_values["native.country"] | |
| def main(): | |
| st.title("Adult Income") | |
| with st.form("questionaire"): | |
| age = # user's input | |
| workclass = # user's input | |
| education = # user's input | |
| Marital_Status = # user's input | |
| occupation = # user's input | |
| relationship = # user's input | |
| race = # user's input | |
| sex = # user's input | |
| hours_per_week = # user's input | |
| native_country = # user's input | |
| # clicked==True only when the button is clicked | |
| clicked = st.form_submit_button("Predict income") | |
| if clicked: | |
| result=model.predict(pd.DataFrame({"age": [age], | |
| "workclass": [workclass], | |
| "education": [EDU_DICT[education]], | |
| "marital.status": [Marital_Status], | |
| "occupation": [occupation], | |
| "relationship": [relationship], | |
| "race": [race], | |
| "sex": [sex], | |
| "hours.per.week": [hours_per_week], | |
| "native.country": [native_country]})) | |
| # Show prediction | |
| # Run main() | |