Skip to content

support mlr3 #13

Description

@simonpcouch

Some work on the mlr3 branch already.

Activity

  1. simonpcouch commented on Jul 18, 2022

    @simonpcouch
    CollaboratorAuthor

    After spending more time with this, I think our best approach here will to wait for more machinery from mlr3 folks before implementing anything more principled. Prediction in a new session works fine, so this shouldn't be a pain point!

    library(mlr3)
    
    task <- tsk("mtcars")
    fit <- lrn("regr.rpart")
      
    # train a model of this learner for a subset of the task
    fit$train(task, row_ids = 1:26)
    
    callr::r(
      function(fit) {
        library(mlr3)
        
        predict(fit, mtcars[27:32,])
      },
      args = list(fit = fit)
    )
    #> [1] 27.71429 16.87368 16.87368 16.87368 16.87368 16.87368

    Created on 2022-07-18 by the reprex package (v2.0.1)

  2. simonpcouch commented on Jul 19, 2022

    @simonpcouch
    CollaboratorAuthor

    Look for the fitted model object slot, these should wrap like caret or parsnip.

  3. sebffischer commented on Jan 24, 2023

    @sebffischer

    I have started a discussion in mlr3 about serialization here: mlr-org/mlr3#891
    If we implement something like this, adding a bundle method for mlr3 Learners should be straightforward! :)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    featurea feature request or enhancement

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions