Training the model instead of training the network for non NN models (Logistic Regression)

Here: https://code.fbi.h-da.de/pse-ai-at-the-edge/ai-at-the-edge/-/blob/a44f95c6738e38a2695d6257346a78b0f5e4a6fa/book/src/AI-Models/Logistic_Regression/Logistic_Regression.md#L46