Nathanotal commited on
Commit
316c194
·
1 Parent(s): ad32c72

add classes

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -71,9 +71,12 @@ def titanic(Pclass, Sex, Age, SibSp, Parch, Fare, Cabin, Embarked):
71
 
72
 
73
  inputs = []
74
- numericalInputs = ["age", "sibsp", "parch", "fare", "pclass"]
75
- worthlessInputs = ["name", "ticket"]
76
- categoricalInputs = ["sex", "embarked", "cabin"]
 
 
 
77
 
78
  for feature in featureLabels:
79
  if feature in numericalInputs:
 
71
 
72
 
73
  inputs = []
74
+ numericalInputs = ["Age", "SibSp", "Parch", "Fare", "Pclass"]
75
+ worthlessInputs = ["Name", "Ticket"]
76
+ categoricalInputs = ["Sex", "Embarked", "Cabin"]
77
+
78
+ featureLabels = ["Pclass", "Name", "Sex", "Age", "SibSp",
79
+ "Parch", "Ticket", "Fare", "Cabin", "Embarked"]
80
 
81
  for feature in featureLabels:
82
  if feature in numericalInputs: