search

LaurentMazare / ocaml-torch

OCaml bindings for PyTorch
Apache License 2.0
412 stars 38 forks source link

Is there a regressor example in there? #37

Closed UnixJunkie closed 3 years ago

UnixJunkie commented 4 years ago

I could not find one. :(

LaurentMazare commented 4 years ago

What do you mean by regressor ? If it's a linear regression that you're after, I think the closest thing is the optimizer test. Let me know if you have any question/want more details.

UnixJunkie commented 4 years ago

Something to predict a scalar, instead of a class label.

LaurentMazare commented 4 years ago

Ok, then the linear regression example above should indeed do this.

LaurentMazare commented 3 years ago

Closing as no further activity in a while, feel free to re-open if useful.