본문 바로가기
Computer Science/Machine Learning

Discriminant analysis(판별 분석) [Discriminant Analysis 이용 원자력발전소 상태 예측] (3)

by BaekDaBang 2024. 3. 31.

 LDA

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

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))

QDA

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

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

print(accuracy_score(y_train,y_train_pred))