随机森林、数据集划分、准确率、混淆矩阵(Python实现)

[8, 8, 8],

[9, 9, 9]]

#输出 x_test每组数据的预测结果的标签值=

print(model.predict(x_test))

#输出的是二维矩阵 ,第i行j列表示 x_test第i行测试数据在每个label上的概率(这李标签共有5个 0,1,2,3,4)

print(model.predict_proba(x_test))

#predict结果:

[2 4 4 4]

#predict_proba结果:

[[0. 0.38 0.45 0.14 0.03]

[0. 0. 0.08 0.16 0.76]

[0. 0. 0.08 0.16 0.76]

[0. 0. 0.08 0.16 0.76]]

Process finished with exit code 0

1.2 案例2

情景:基于相亲网站男方提供的个人基本资料作为输入,以女方是否相亲做标签做训练,用训练出的模型预测女方是否相亲。

#导包=

from sklearn.ensemble import RandomForestClassifier

#“年龄”, “身高”, “年收入”, ‘学历’(0:大专, 1:本科, 2:硕士)

X = [

[25, 179, 15, 0],

[33, 190, 19, 0],

[28, 180, 18, 2],

[25, 178, 18, 2],

[46, 100, 100, 2],

[40, 170, 170, 1],

[34, 174, 20, 2],

[36, 181, 55, 1],

[35, 170, 25, 2],

[30, 180, 35, 1],

[28, 174, 30, 1],

[29, 176, 36, 1],

]

#=有否相亲 0:N 1:Y=========

y = [0, 1, 1, 1, 0, 0, 1, 0, 1, 1, 0, 1]

#现在我们把训练数据,和对应的分类放入分类器中进行训练

model = RandomForestClassifier().fit(X, y)

  • 22
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,让我来回答你的问题。你想知道如何使用Python代码对澳大利亚信贷批准数据集进行线性逻辑回归和非线性随机森林分类,并进行特征选择,最后使用准确度、召回率和混淆矩阵等指标来评估模型的性能,是吗? 以下是一个大致的流程: 1. 加载数据集并进行数据预处理 首先,你需要加载数据集并进行一些简单的数据预处理,如缺失值填充和特征标准化。以下是一个示例代码: ``` python import pandas as pd from sklearn.preprocessing import StandardScaler # 加载数据集 df = pd.read_csv('australian.csv') # 填充缺失值 df.fillna(df.mean(), inplace=True) # 特征标准化 scaler = StandardScaler() X = scaler.fit_transform(df.iloc[:, :-1]) y = df.iloc[:, -1].values ``` 2. 构建线性逻辑回归模型并进行特征选择 你可以使用Scikit-learn库中的逻辑回归算法来构建线性分类模型,并使用L1或L2正则化来进行特征选择。以下是一个示例代码: ``` python from sklearn.linear_model import LogisticRegression from sklearn.feature_selection import SelectFromModel # 构建逻辑回归模型并进行特征选择 lr = LogisticRegression(penalty='l1', solver='liblinear', random_state=42) selector = SelectFromModel(lr) X_new = selector.fit_transform(X, y) ``` 3. 构建非线性随机森林模型并进行特征选择 你可以使用Scikit-learn库中的随机森林算法来构建非线性分类模型,并使用基于特征重要性的方法来进行特征选择。以下是一个示例代码: ``` python from sklearn.ensemble import RandomForestClassifier from sklearn.feature_selection import SelectFromModel # 构建随机森林模型并进行特征选择 rf = RandomForestClassifier(n_estimators=100, random_state=42) selector = SelectFromModel(rf) X_new = selector.fit_transform(X, y) ``` 4. 评估模型性能 最后,你可以使用准确度、召回率和混淆矩阵等指标来评估模型的性能。以下是一个示例代码: ``` python from sklearn.metrics import confusion_matrix, accuracy_score, recall_score from sklearn.model_selection import train_test_split # 划分训练集和测试集 X_train, X_test, y_train, y_test = train_test_split(X_new, y, test_size=0.3, random_state=42) # 构建逻辑回归模型并评估 lr = LogisticRegression(random_state=42) lr.fit(X_train, y_train) y_pred_lr = lr.predict(X_test) print("逻辑回归模型的混淆矩阵:\n", confusion_matrix(y_test, y_pred_lr)) print("逻辑回归模型的准确度:", accuracy_score(y_test, y_pred_lr)) print("逻辑回归模型的召回率:", recall_score(y_test, y_pred_lr)) # 构建随机森林模型并评估 rf = RandomForestClassifier(n_estimators=100, random_state=42) rf.fit(X_train, y_train) y_pred_rf = rf.predict(X_test) print("随机森林模型的混淆矩阵:\n", confusion_matrix(y_test, y_pred_rf)) print("随机森林模型的准确度:", accuracy_score(y_test, y_pred_rf)) print("随机森林模型的召回率:", recall_score(y_test, y_pred_rf)) ``` 请注意,以上代码仅是一个示例,你可能需要根据具体情况进行更改和调整。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值