Commit 37fa924e authored by Susheel Busi's avatar Susheel Busi
Browse files

fixed PCA embedding

parent 08232b99
......@@ -75,8 +75,8 @@ def create_kmeans_data(data, labels):
def PCA_embedding(X, n):
pca = PCA(n_components=n)
X_freq = FREQ_transform(X)
return pca.fit_transform(X_freq)
# X_freq = FREQ_transform(X)
return pca.fit_transform(X)
def KMEANS_embedding(X, km):
"""Transforms the counts contained in the data X into K-means clustered frequencies."""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment