StandardScaler
This commit is contained in:
parent
de7b569a89
commit
0a1dd766b6
@ -5,7 +5,7 @@ import seaborn as sns
|
|||||||
|
|
||||||
from numpy.random import choice, seed
|
from numpy.random import choice, seed
|
||||||
from sklearn.decomposition import PCA
|
from sklearn.decomposition import PCA
|
||||||
from sklearn.preprocessing import Normalizer
|
from sklearn.preprocessing import StandardScaler
|
||||||
|
|
||||||
seed(42)
|
seed(42)
|
||||||
|
|
||||||
@ -44,7 +44,7 @@ def is_finished(old_medoids, new_medoids):
|
|||||||
def kmedoids(num_samples, num_clusters):
|
def kmedoids(num_samples, num_clusters):
|
||||||
df = pd.read_csv('CC GENERAL.csv', index_col='CUST_ID')
|
df = pd.read_csv('CC GENERAL.csv', index_col='CUST_ID')
|
||||||
df = df[:num_samples].fillna(0)
|
df = df[:num_samples].fillna(0)
|
||||||
df_scaled = pd.DataFrame(Normalizer().fit_transform(df))
|
df_scaled = pd.DataFrame(StandardScaler().fit_transform(df))
|
||||||
|
|
||||||
# initialize medoids (at random)
|
# initialize medoids (at random)
|
||||||
medoids = initialize_medoids(num_medoids=num_clusters, data=df_scaled)
|
medoids = initialize_medoids(num_medoids=num_clusters, data=df_scaled)
|
||||||
|
Loading…
Reference in New Issue
Block a user