Machine Learning - Python.

Olá. Estou estudante de física e faço um projeto de IC na Matemática. Preciso desenvolver um código em Python para classificar os conjuntos assintóticos de aplicações polinomiais usando um algoritmo criado pela minha orientadora.

Ela me deu duas possibilidades, utilizar o algoritmo em si (creio ser mais difícil) ou utilizar as classificações já feitas por ela. Ela levantou que qualquer equação, dependendo da forma, podem se encaixar em 9 tipos diferentes de classificações (plano, intersecção de planos, paraboloides, etc.).

Pensei em usar Machine Learning, porém não sei muito bem como fazer isso. Já utilizei Python, C e Java, porém deve ser desenvolvido em Python. Gostaria de uma ajuda em como poderia desenvolver esse código, preciso que a pessoa digite uma equação e que o código classifique e dê a equação da superfície.