from sklearn.datasets import load_iris from sklearn.model_selection import train_test_split
data = load_iris(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(data[0], data[1], test_size=0.3, random_state=0)
from sklearn.svm import SVC from sklearn.tree import DecisionTreeClassifier from sklearn.ensemble import RandomForestClassifier from sklearn.neural_network import MLPClassifier
svm = SVC() dtc = DecisionTreeClassifier() rfc = RandomForestClassifier() mlpc = MLPClassifier()
svm.fit(X_train, y_train) svm_score_train = svm.score(X_train, y_train) svm_score_test = svm.score(X_test, y_test)
scores_train = [svm_score_train, dtc_score_train, rfc_score_train, mlpc_score_train] scores_test = [svm_score_test, dtc_score_test, rfc_score_test, mlpc_score_test]
max_score_train = max(scores_train) max_score_test = max(scores_test)
if max_score_test > max_score_train: best_clf = classifiers[scores_test.index(max_score_test)] else: best_clf = classifiers[scores_train.index(max_score_train)]
print("The best
下一篇:比较计数列未被识别