Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Correct datatype for multi-output forests #123

Open
fipelle opened this issue Mar 27, 2023 · 3 comments
Open

Correct datatype for multi-output forests #123

fipelle opened this issue Mar 27, 2023 · 3 comments

Comments

@fipelle
Copy link

fipelle commented Mar 27, 2023

Hi,

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

@cstjean
Copy link
Owner

cstjean commented Mar 27, 2023

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

@fipelle
Copy link
Author

fipelle commented Mar 29, 2023

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
Copy link
Owner

cstjean commented Mar 30, 2023

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants