Skip to content
This repository has been archived by the owner on Jul 31, 2021. It is now read-only.

Prediction is crashed when restart r #16

Open
nguyenngocbinh opened this issue Jul 28, 2021 · 0 comments
Open

Prediction is crashed when restart r #16

nguyenngocbinh opened this issue Jul 28, 2021 · 0 comments

Comments

@nguyenngocbinh
Copy link

nguyenngocbinh commented Jul 28, 2021

library(mlr3)
library(mlr3learners.lightgbm)
task = mlr3::tsk("iris")
learner = mlr3::lrn("classif.lightgbm", objective = "multiclass")

learner$param_set$values = mlr3misc::insert_named(
  learner$param_set$values,
  list(
    "early_stopping_round" = 10,
    "learning_rate" = 0.1,
    "seed" = 17L,
    "metric" = "multi_logloss",
    "num_iterations" = 100,
    "num_class" = 3
  )
)

learner$train(task, row_ids = 1:120)
save(learner, task, file = "test_lightgbm.RData")

I restart Rstudio (press Ctrl + Shift + F10)

load library again

library(mlr3)
library(mlr3learners.lightgbm)
rm(list = ls())
load("test_lightgbm.RData")
predictions = learner$predict(task, row_ids = 121:150) # Crash in this  -----------
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant