search

cstjean / ScikitLearn.jl

Julia implementation of the scikit-learn API https://cstjean.github.io/ScikitLearn.jl/dev/
Other
547 stars 75 forks source link

Correct datatype for multi-output forests #123

Open fipelle opened 1 year ago

fipelle commented 1 year ago

Hi,

I am trying to replicate this in Julia. What is the correct datatype for y? Would Vector{Vector{Float64}} work as intended?

cstjean commented 1 year ago

Give it a try. That Python code looks like a matrix to me, so I'd try Matrix{Float64}.

fipelle commented 1 year ago

Thanks, do you know if it is possible to pass a custom loss function / split criterion? I would like to weight the targets if possible.

cstjean commented 1 year ago

I think it should work just like in Python, and you can pass an arbitrary Julia function. Give it a try and post the error message.