Nathanotal commited on
Commit
375f1b4
·
1 Parent(s): a17fb98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -61,18 +61,18 @@ featureLabels = ["Pclass", "Name", "Sex", "Age", "SibSp",
61
  inputs = []
62
  numericalInputs = ["Age", "SibSp", "Parch", "Fare"]
63
  # Maybe move cabin to categorical
64
- worthlessInputs = ["Name", "Ticket", "Cabin"]
65
  categoricalInputs = ["Sex", "Embarked", "Pclass"]
66
 
67
  columnHeaders = ["Pclass", "Sex", "Age", "SibSp",
68
- "Parch", "Fare", "Embarked"]
69
 
70
 
71
  def titanic(Pclass, Sex, Age, SibSp, Parch, Fare, Embarked):
72
 
73
  # Create a dataframe from the input values
74
  input_variables = pd.DataFrame(
75
- [[Pclass, Sex, Age, SibSp, Parch, Fare, Embarked]], columns=columnHeaders)
76
  df = prepare_for_write(input_variables)
77
 
78
  # Save first row as a numpy array
 
61
  inputs = []
62
  numericalInputs = ["Age", "SibSp", "Parch", "Fare"]
63
  # Maybe move cabin to categorical
64
+ worthlessInputs = ["Name", "Ticket", "Cabin", "Title"]
65
  categoricalInputs = ["Sex", "Embarked", "Pclass"]
66
 
67
  columnHeaders = ["Pclass", "Sex", "Age", "SibSp",
68
+ "Parch", "Fare", "Embarked", "Title"] # Todo: remove title
69
 
70
 
71
  def titanic(Pclass, Sex, Age, SibSp, Parch, Fare, Embarked):
72
 
73
  # Create a dataframe from the input values
74
  input_variables = pd.DataFrame(
75
+ [[Pclass, Sex, Age, SibSp, Parch, Fare, Embarked, "Mr"]], columns=columnHeaders)
76
  df = prepare_for_write(input_variables)
77
 
78
  # Save first row as a numpy array