Comment by alegrey91

Let me know what do you think in case you decide to use it. The model is still far from being perfect, but it gives decent results considering I'm new to the ML topic.