Linear Discriminant Analysis (LDA) | ลดมิติพร้อมแยกคลาส

import numpy as np
import matplotlib.pyplot as plt
import japanize_matplotlib
from sklearn.datasets import make_classification
from sklearn.discriminant_analysis import LinearDiscriminantAnalysis

X, y = make_classification(
    n_samples=300,
    n_features=5,
    n_informative=3,
    n_classes=3,
    n_clusters_per_class=1,
    random_state=0,
)

lda = LinearDiscriminantAnalysis(n_components=2)
X_proj = lda.fit_transform(X, y)

plt.figure(figsize=(6, 5))
plt.scatter(X_proj[:, 0], X_proj[:, 1], c=y, cmap="tab10")
plt.title("ลดมิติด้วย LDA (2 components)")
plt.xlabel("LD1")
plt.ylabel("LD2")
plt.grid(alpha=0.2)
plt.tight_layout()
plt.show()