unknown commited on
Commit
460dd35
·
1 Parent(s): 8e53334
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -19,15 +19,16 @@ model = joblib.load(model_dir + "/titan_model.pkl")
19
  def titan(pclass, sex, age, fare, famliy):
20
  input_list = []
21
  if pclass == "First Class":
22
- input_list.append(int(1))
23
  elif pclass == "Second Class":
24
- input_list.append(int(2))
25
  else:
26
- input_list.append(int(3))
 
27
  if sex == "Male":
28
- input_list.append(int(0))
29
  else:
30
- input_list.append(int(1))
31
  input_list.append(sex)
32
  input_list.append(age)
33
  input_list.append(fare)
 
19
  def titan(pclass, sex, age, fare, famliy):
20
  input_list = []
21
  if pclass == "First Class":
22
+ input_list.append(1)
23
  elif pclass == "Second Class":
24
+ input_list.append(2)
25
  else:
26
+ input_list.append(3)
27
+
28
  if sex == "Male":
29
+ input_list.append(0)
30
  else:
31
+ input_list.append(1)
32
  input_list.append(sex)
33
  input_list.append(age)
34
  input_list.append(fare)