ссылка на столбец в xgb.DMatrix для прогнозирования модели - PullRequest
Новая
       30

ссылка на столбец в xgb.DMatrix для прогнозирования модели

0 голосов
/ 15 мая 2019

Я хотел бы сослаться на столбец в xgb.DMatrix и использовать его в условии if. Можно ли ссылаться на элементы в столбце в xgb.DMatrix? (Я использую пакет xgboost)

Например: 

mat1_xgb[date_created<latest_date,]

, где 

 latest_date is some date in numeric format
 mat1_xgb$date_created is a column of dates in numeric formats

Пример

В коде r: 

Matrix_1:
| date_created | val1 | date_closed |
|  17912       |2.3   |  17914      |
| 17913        |2.  5 |  17915      |
| 17932        |2.1   |  17941      |

Затем создайте цикл for, в котором модель xgboost запускается с использованием только дат, предшествующих дате приращения: 

for(i in seq(2,dim(Matrix_1)[1],1)){

   latest_date = Matrix_1[2,date_created]
   xgb1_learn = xgboost(data = Matrix_1[date_closed<latest_date], params = params, etc)
   data_table_1[i:i+50] = predict(xgb1_learn, Matrix_1[i:i+50])
}

Таким образом, мы обучаем модель только на прошлых точках данных

...