Classifier Comparison 2.py

#Train Models and get score
train_scores=[]
test_scores=[]
for model in model_list:
    model.fit(X_train, y_train)
    train_scores.append(model.score(X_train, y_train))
    test_scores.append(model.score(X_test, y_test))
 
scores_df= pd.DataFrame(
    {'Model Name': ['tree','logreg','knn','naive_b','forest','gbrt','mlp','svc'],
     'Train Score': train_scores,
     'Test Score': test_scores
    })
 
scores_df.sort_values(by=['Test Score'],ascending=False,inplace=True)    
print(scores_df)