I have the following piece of code, and wonder if fit() functionality is doing things correctly or not. In particular, I have this little check of counting the times train_s
