[Dimension Reduction] NMF 변환

 

 

 

NMF¶

 

In [1]:

from sklearn.decomposition import NMF
from sklearn.datasets import load_iris
import matplotlib.pyplot as plt
%matplotlib inline

 

 

• iris

 

In [2]:

iris = load_iris()
iris_ftrs = iris.data

 

 

• NMF 변환

 

In [12]:

nmf = NMF(n_components=2, max_iter=1000)
nmf.fit(iris_ftrs)
iris_nmf = nmf.transform(iris_ftrs)

 

 

• NMF 변환된 입력 데이터 값 시각화

 

In [13]:

plt.scatter(x=iris_nmf[:,0], y= iris_nmf[:,1], c= iris.target)
plt.xlabel('NMF Component 1')
plt.ylabel('NMF Component 2')

 

Out[13]:

Text(0, 0.5, 'NMF Component 2')