机器学习模型计算accuracy

一：分类问题

from sklearn.metrics import accuracy_score from sklearn.model_selection import train_test_split X_train, X_test, y_train, y_test = train_test_split(train_x, train_y, test_size=0.33, random_state=123) model.fit(X_train, y_train) x_pred = xgbr.predict(X_test) accuracy_score(y_test, y_pred)

二：回归问题

from sklearn.metrics import mean_squared_error from sklearn.model_selection import train_test_split X_train, X_test, y_train, y_test = train_test_split(train_x, train_y, test_size=0.33, random_state=123) model.fit(X_train, y_train) x_pred = xgbr.predict(X_test) mean_squared_error(y_test, y_pred)

参考