Skip to content

SoleXplorer compatibility #67

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

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

PasoStudio73
Copy link
Member

when performing a sprinkle to the model, to obtain a Sole compatible model, and prediction, the function sprinkle creates a Sole model but actual predictions aren't written in Sole struct (under the struct 'info') instead are passed as a second parameter.
SoleXplorer expets to find predictions inside the model, under info struct.

in order to have sprinkle properly working for SoleXplorer, i've added a function in mlj fit interface: 'translate_model', called by sprinkle in place of 'translate_function' that I kept as it is for safety.
the difference between translate_model and translate_function is that the first one accepts also predictions as a vector of string.
then, call translate function to convert the tree.
I also make a little change in translate function: since it return a wrong supporting predictions, if a vector of predictions is passed, it will be merged.

@PasoStudio73 PasoStudio73 linked an issue Apr 18, 2025 that may be closed by this pull request
@PasoStudio73 PasoStudio73 requested a review from giopaglia April 18, 2025 16:17
@PasoStudio73 PasoStudio73 self-assigned this Apr 18, 2025
@PasoStudio73 PasoStudio73 added the suggested change Small review suggestions, such as typo fixes label Apr 18, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
suggested change Small review suggestions, such as typo fixes
Projects
Status: No status
Development

Successfully merging this pull request may close these issues.

SoleXplorer compatibility
1 participant