extend py plugin for machine learning - View it on GitHub