Получить доступ к регрессионной модели поезда - PullRequest
3 голосов
/ 15 января 2020

Я получил упражнение, в котором мне нужно обучить модель линейной регрессии и получить некоторую информацию о модели:

  • линейное соотношение между выбранной переменной и другими переменными
  • какие переменные важны для модели
  • значимость

Создать модель с помощью функции lm легко, поэтому я могу интерпретировать ее с помощью сводки (мода).

mod <- lm(cars$height ~ ., data = cars)

Summary () - MEthod возвращает все: r-квадрат, коэффициенты, p-значение, значимость ...

Но когда я тренирую мою модель, как:

library(mlr)

lrn = makeLearner("regr.ksvm")

mod = train(learner = lrn, task = task)

pred = predict(object = mod, newdata = test)

performance(pred = pred, measures = list(mse, arsq))

Я просто получаю mse и r-squareZd. Как получить другую информацию, такую ​​как значимость, важные переменные ... Есть ли шанс получить доступ к этому моду?

Спасибо за помощь

1 Ответ

4 голосов
/ 15 января 2020
library(mlr)
#> Loading required package: ParamHelpers
#> 'mlr' is in maintenance mode since July 2019. Future development
#> efforts will go into its successor 'mlr3' (<https://mlr3.mlr-org.com>).

lrn = makeLearner("regr.lm")

mod = train(learner = lrn, task = bh.task)

getLearnerModel(mod)
#> 
#> Call:
#> stats::lm(formula = f, data = d)
#> 
#> Coefficients:
#> (Intercept)         crim           zn        indus        chas1          nox  
#>   3.646e+01   -1.080e-01    4.642e-02    2.056e-02    2.687e+00   -1.777e+01  
#>          rm          age          dis          rad          tax      ptratio  
#>   3.810e+00    6.922e-04   -1.476e+00    3.060e-01   -1.233e-02   -9.527e-01  
#>           b        lstat  
#>   9.312e-03   -5.248e-01

Создано в 2020-01-15 пакетом представ (v0.3.0.9001)

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...