Update train.py

train.py

@@ -27,7 +27,7 @@ model.add(Dropout(0.5)) # dropout makes ___ task harder __ removing ____ informa
 model.add(Dense(512, activation="relu"))
 model.add(Dense(512, activation="relu"))
 model.add(Dense(256, activation="relu"))
-model.add(Dense(dset_size, activation="linear")) # TBH it doesn't matter that much what activation function to use, just linear does nothing at all to the output, that might be something like softmax but i'll test that later
+model.add(Dense(dset_size, activation="linear")) # TBH it doesn't matter that much what activation function to use IN THIS CASE, IN THIS LINE (in others it might be a really big deal), just linear does nothing at all to the output, that might be something like softmax but i'll test that later
 
 X = [] # we're loading the training data into input X
 y = [] # and output y