Python machine learning backend for trifi