PerryXDeng 5 years ago
parent
commit
c3cb855519
1 changed files with 25 additions and 0 deletions
  1. +25
    -0
      data_exploration/wellness_cluster_test.py

+ 25
- 0
data_exploration/wellness_cluster_test.py View File

@ -0,0 +1,25 @@
import sklearn.cluster as cluster
import pandas as pd
from matplotlib import pyplot as plt
def find_kmeans(mat, k, first, second):
km = cluster.KMeans(n_clusters=k)
km.fit(mat)
# Plot sse against k
plt.figure(figsize=(6, 6))
plt.xlabel('Metric: ' + first)
plt.ylabel('Metric: ' + second)
plt.scatter(mat[:, 0], mat[:, 1], c=km.labels_, cmap='rainbow')
plt.show()
# Read csv in
df = pd.read_csv('../data_preparation/cleaned/time_series_normalized_wellness_menstruation.csv')
# Specify what things you want
df = df[["normFatigue", "normSleepQuality"]]
# values, num clusters, axis labelsg
find_kmeans(df.values, 2, "normFatigue", "normSleepQuality")

Loading…
Cancel
Save