기본 지도 학습 알고리즘들로지스틱 회귀 (Logistic Regression)scikit-learn으로 로지스틱 회귀 쉽게하기
fit에서의 오류
조회 701
좋아요 0
2021년 7월 23일
흔들방구
0
채택된 답변 수 2,645
댓글 3개
댓글 3개
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import mean_squared_error
import pandas as pd
import numpy as np
iris_data = load_iris()
# print(iris_data.DESCR) #50개는 setosa 50개는Versico 50개는Virginica
X=pd.DataFrame(iris_data.data , columns=iris_data.feature_names)
y=pd.DataFrame(iris_data.target, columns=["class"])
X_train, X_test , y_train, y_test = train_test_split(X ,y, test_size=0.2 , random_state= 5)
y_train=y_train.values.ravel()
model=LogisticRegression(solver='saga', max_iter=2000)
#solver= 최적화에 사용할 알고리즘(Saga , sag ..etc) max_iter = 최적화시 반복 횟수
model.fit(X_train, y_train)
model.predict(X_test)
a=model.score(X_test, y_test)
print(a)