I have been using the Py-earth model, the url can be found here:
And following this turorial here :
I have managed to train:
model = Earth(max_d