search

scikit-learn-contrib / py-earth

A Python implementation of Jerome Friedman's Multivariate Adaptive Regression Splines
http://contrib.scikit-learn.org/py-earth/
BSD 3-Clause "New" or "Revised" License
457 stars 121 forks source link

documentation of bivariate responses? #158

Closed ghost closed 7 years ago

ghost commented 7 years ago

This isn't really an issue. It's more like a question, and it really is much more suitable for an open discussion or help forum for py-earth than for here. If there is such a thing, could someone point me to it, and then this Issue can be closed.

Otherwise, I was looking for an example of how one can set up a case for a bivariate response model for py-earth. Is it as simple as submitting an n-by-2 numpy matrix for the second argument to the fit method? From the API, I assume so. But where do I find what's returned in the case of multivariate y, that is, with p > 1?

ghost commented 7 years ago

Figured it out from code and such.

jcrudy commented 7 years ago

@empirical-bayesian Excellent. If anyone else comes across this thread before I get around to writing better documentation, a good place to look in in examples/plot_sine_wave_2d.py.