matsammut commited on
Commit
0e84b81
·
verified ·
1 Parent(s): 2aca552

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,7 +38,7 @@ def predict_rf(age, workclass, education, marital_status, occupation, relationsh
38
  # "age": [age], "workclass":[workclass], "educational-num":[education], "marital-status":[marital_status], "occupation":[occupation],
39
  # "relationship":[relationship], "race":[race], "gender":[gender], "capital-gain":[capital_gain], "capital-loss":[capital_loss],
40
  # "hours-per-week":[hours_per_week], "native-country":[native_country]}
41
- columns = { "0":[0],
42
  "age": [age], "workclass":[workclass], "educational-num":[education], "occupation":[occupation],
43
  "race":[race], "gender":[gender], "capital-gain":[capital_gain], "capital-loss":[capital_loss],
44
  "hours-per-week":[hours_per_week], "native-country":[native_country]}
@@ -58,7 +58,7 @@ def predict_hb(age, workclass, education, marital_status, occupation, relationsh
58
  # "age": [age], "workclass":[workclass], "educational-num":[education], "marital-status":[marital_status], "occupation":[occupation],
59
  # "relationship":[relationship], "race":[race], "gender":[gender], "capital-gain":[capital_gain], "capital-loss":[capital_loss],
60
  # "hours-per-week":[hours_per_week], "native-country":[native_country]}
61
- columns = { "0":[0],
62
  "age": [age], "workclass":[workclass], "educational-num":[education], "occupation":[occupation],
63
  "race":[race], "gender":[gender], "capital-gain":[capital_gain], "capital-loss":[capital_loss],
64
  "hours-per-week":[hours_per_week], "native-country":[native_country]}
 
38
  # "age": [age], "workclass":[workclass], "educational-num":[education], "marital-status":[marital_status], "occupation":[occupation],
39
  # "relationship":[relationship], "race":[race], "gender":[gender], "capital-gain":[capital_gain], "capital-loss":[capital_loss],
40
  # "hours-per-week":[hours_per_week], "native-country":[native_country]}
41
+ columns = {
42
  "age": [age], "workclass":[workclass], "educational-num":[education], "occupation":[occupation],
43
  "race":[race], "gender":[gender], "capital-gain":[capital_gain], "capital-loss":[capital_loss],
44
  "hours-per-week":[hours_per_week], "native-country":[native_country]}
 
58
  # "age": [age], "workclass":[workclass], "educational-num":[education], "marital-status":[marital_status], "occupation":[occupation],
59
  # "relationship":[relationship], "race":[race], "gender":[gender], "capital-gain":[capital_gain], "capital-loss":[capital_loss],
60
  # "hours-per-week":[hours_per_week], "native-country":[native_country]}
61
+ columns = {
62
  "age": [age], "workclass":[workclass], "educational-num":[education], "occupation":[occupation],
63
  "race":[race], "gender":[gender], "capital-gain":[capital_gain], "capital-loss":[capital_loss],
64
  "hours-per-week":[hours_per_week], "native-country":[native_country]}