Pamudu13 commited on
Commit
2809785
·
verified ·
1 Parent(s): ac880b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ models = {
37
  }
38
 
39
  # Choose and train a model
40
- model_choice = "Decision Tree" # Default model
41
  obj = models[model_choice]
42
  obj.fit(x_train, y_train)
43
  y_pred = obj.predict(x_test)
 
37
  }
38
 
39
  # Choose and train a model
40
+ model_choice = "Naive Bayes" # Default model
41
  obj = models[model_choice]
42
  obj.fit(x_train, y_train)
43
  y_pred = obj.predict(x_test)