본문 바로가기
Computer Science/Machine Learning

Discriminant analysis(판별 분석) [Discriminant Analysis 이용 당뇨병 환자 예측] (4)

by BaekDaBang 2024. 3. 31.

1. Dataset

Train

 

Test

x_train = train.drop(['Diabetes'], axis=1)
y_train = train['Diabetes']
x_test = test

 

2. LDA

# LDA
from sklearn.discriminant_analysis import LinearDiscriminantAnalysis
from sklearn.metrics import accuracy_score

cld = LinearDiscriminantAnalysis()
cld.fit(x_train, y_train)
y_train_pred = cld.predict(x_train)
y_test_cld_pred = cld.predict(x_test)

print(accuracy_score(y_train, y_train_pred))

 

3. QDA

# QDA
from sklearn.discriminant_analysis import QuadraticDiscriminantAnalysis
from sklearn.metrics import accuracy_score

cqd=QuadraticDiscriminantAnalysis()
cqd.fit(x_train, y_train)
y_train_pred = cqd.predict(x_train)
y_test_cqd_pred = cqd.predict(x_test)

print(accuracy_score(y_train, y_train_pred))