Esempio n. 1
0
from sklearn.discriminant_analysis import LinearDiscriminantAnalysis as LDA
import plot_decision_regions as pdr
import wine_data
from sklearn.linear_model import LogisticRegression
import matplotlib.pyplot as plt

w = wine_data.WineDataSets()
lda = LDA(n_components = 2)
X_train_lda = lda.fit_transform(w.X_train_std, w.y_train)

lr = LogisticRegression()
lr = lr.fit(X_train_lda, w.y_train)
pdr.plot_decision_regions(X_train_lda, w.y_train, classifier = lr)
plt.xlabel('LD 1')
plt.ylabel('LD 2')
plt.legend(loc = 'lower left')
plt.show()

X_test_lda = lda.transform(w.X_test_std)
pdr.plot_decision_regions(X_test_lda, w.y_test, classifier = lr)
plt.xlabel('LD 1')
plt.ylabel('LD 2')

plt.legend(loc = 'lower left')
plt.show()
Esempio n. 2
0
import matplotlib.pyplot as plt
from sklearn.decomposition import PCA
import wine_data as wd
from sklearn.linear_model import LogisticRegression
import plot_decision_regions as pdr

pca = PCA(n_components=2)
lr = LogisticRegression()
w = wd.WineDataSets()
X_train_pca = pca.fit_transform(w.X_train_std)
X_test_pca = pca.transform(w.X_test_std)
lr.fit(X_train_pca, w.y_train)
pdr.plot_decision_regions(X_train_pca, w.y_train, classifier=lr)
plt.xlabel('PC1')
plt.xlabel('PC2')
plt.legend(loc='lower left')
plt.show()

pdr.plot_decision_regions(X_test_pca, w.y_test, classifier=lr)
plt.xlabel('PC1')
plt.xlabel('PC2')
plt.legend(loc='lower left')
plt.show()