add some comments

lazyprogrammer · Jun 22, 2020 · 4e5e20e · 4e5e20e
1 parent 069b5d4
commit 4e5e20e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/unsupervised_class/neural_kmeans.py b/unsupervised_class/neural_kmeans.py
@@ -4,21 +4,24 @@
 from sklearn.preprocessing import StandardScaler
 
 
+# get the data and standardize it
 X = get_simple_data()
 scaler = StandardScaler()
 X = scaler.fit_transform(X)
 
+# get shapes
 N, D = X.shape
 K = 3
 
+# initialize parameters
 W = np.random.randn(D, K)
 
-
+# set hyperparameters
 n_epochs = 100
 learning_rate = 0.001
 losses = []
 
-
+# training loop
 for i in range(n_epochs):
   loss = 0
   for j in range(N):