clf.predict_proba(X_train)[:, 1]是什么意思？

  y_train_prob = clf.predict_proba(X_train)[:, 1]

这行代码让我疑惑，它究竟是什么意思呢？

查资料外做实验，搞懂了。

print(clf.predict(x_test))

# 返回预测标签

就是预测值是哪些

print(clf.predict_proba(x_test))

# 返回预测属于某标签的概率

就是属于某类预测值的概率是多少

而对于[:,1]

是取二维数组中第二维的所有数据

举个例子：

 y_train_prob = clf.predict_proba(X_train)[:, 1]print("clf.predict_proba(X_train)")print(clf.predict_proba(X_train))print("clf.predict_proba(X_train)[:, 1]")print(y_train_prob)

输出是：

含义是：预测X_rain[1. 0. 1. ... 0. 1. 0.]的标签是0的概率为0.94005474 ，为1的概率为0.05994526

clf.predict_proba(X_train)[:, 1]是什么意思？相关推荐

画ROC为什么clf.predict_proba(X_train)[:, 1] 而不是[:,0]?
y_train_prob = clf.predict_proba(X_train)[:, 1] 这行代码让我疑惑,它究竟是什么意思呢? print(clf.predict(x_test)) # 返回预 ...
关于XGBoost训练数据clf.fit(x_train, y_train)时报错
通过查阅博客发现这是新版本XGBoost常见错误只要把新版本的卸载了下回老版本的就好了 pip uninstall xgboost pip install xgboost==0.90
金融贷款逾期的模型构建6——特征选择
文章目录一.IV值 1.概述 2.IV计算 (1)WOE (2)IV 计算二.实现 0.相关模块 1.IV值 2.Random Forest 3.特征合并 4.模型构建 5.模型评估数据传送门( ...
机器学习算法（二）：基于决策数的分类预测
1.决策树决策树的核心思想是基于树结构对数据进行划分,这种思想是人类处理问题时的本能方法. 决策树的主要优点: 具有很好的解释性,模型可以生成可以理解的规则. 可以发现特征的重要程度. 模型的计算复 ...
机器学习笔记II: 决策树
决策树的介绍决策树是一种常见的分类模型,在金融分控.医疗辅助诊断等诸多行业具有较为广泛的应用.决策树的核心思想是基于树结构对数据进行划分,这种思想是人类处理问题时的本能方法.例如在婚恋市场中,女方通 ...
机器学习笔记I: 基于逻辑回归的分类预测
逻辑回归的介绍逻辑回归(Logistic regression,简称LR)虽然其中带有"回归"两个字,但逻辑回归其实是一个分类模型,并且广泛应用于各个领域之中.虽然现在深度学习相 ...
鸢尾花分类_基于鸢尾花（iris）数据集的逻辑回归分类实践
今天是学习逻辑回归的第三天,不得不说还是挺难的.像我这种小白完全不能在一个小时内搞懂.我仿真这段代码时出现了错误,我还在纠错中,课业繁重,还望见谅. 接下来是代码段 Step1:库函数导入 ## 基础 ...
机器学习笔记-基于逻辑回归的分类预测
天池学习笔记:AI训练营机器学习-阿里云天池基于逻辑回归的分类预测 1 逻辑回归的介绍和应用 1.1 逻辑回归的介绍逻辑回归(Logistic regression,简称LR)虽然其中带有&quo ...
python逻辑回归训练预测_[Python] 机器学习笔记基于逻辑回归的分类预测
导学问题什么是逻辑回归(一),逻辑回归的推导(二 3),损失函数的推导(二 4) 逻辑回归与SVM的异同逻辑回归和SVM都用来做分类,都是基于回归的概念 SVM的处理方法是只考虑 support ...

clf.predict_proba(X_train)[:, 1]是什么意思？

clf.predict_proba(X_train)[:, 1]是什么意思？相关推荐

最新文章

热门文章