python逻辑回归库
时间: 2023-11-19 08:57:44 浏览: 103
python实现的逻辑回归程序
Python中有很多逻辑回归库,其中比较常用的是scikit-learn库中的LogisticRegression模块。下面是一个简单的例子:
```python
from sklearn.linear_model import LogisticRegression
from sklearn.datasets import load_iris
# 加载数据集
iris = load_iris()
X = iris.data[:, :2] # 只选取前两个特征
y = iris.target
# 训练模型
clf = LogisticRegression(random_state=0, solver='lbfgs', multi_class='multinomial').fit(X, y)
# 预测新数据
print(clf.predict([[5.1,3.5]])) # 输出:[0]
```
在这个例子中,我们使用了scikit-learn库中的LogisticRegression模块来训练一个逻辑回归模型,并使用该模型对新数据进行了预测。
阅读全文