Catherine Breslin commited on
Commit
0ad84e1
1 Parent(s): 2486cd2

Clustering

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ from scipy.spatial.distance import cosine
6
  import numpy as np
7
  import seaborn as sns
8
  import matplotlib.pyplot as plt
 
9
 
10
  def cluster_examples(messages, embed, nc=3):
11
  km = KMeans(
 
6
  import numpy as np
7
  import seaborn as sns
8
  import matplotlib.pyplot as plt
9
+ from sklearn.cluster import KMeans
10
 
11
  def cluster_examples(messages, embed, nc=3):
12
  km = KMeans(
requirements.txt CHANGED
@@ -6,3 +6,4 @@ scipy
6
  numpy
7
  seaborn
8
  matplotlib
 
 
6
  numpy
7
  seaborn
8
  matplotlib
9
+ sklearn