gpb.Dataset.set.categorical: Set categorical feature of 'gpb.Dataset'

Description Usage Arguments Value Examples

View source: R/gpb.Dataset.R

Description

Set the categorical features of an gpb.Dataset object. Use this function to tell GPBoost which features should be treated as categorical.

Usage

1
gpb.Dataset.set.categorical(dataset, categorical_feature)

Arguments

dataset

object of class gpb.Dataset

categorical_feature

categorical features. This can either be a character vector of feature names or an integer vector with the indices of the features (e.g. c(1L, 10L) to say "the first and tenth columns").

Value

the dataset you passed in

Examples

1
2
3
4
5
6
7
data(agaricus.train, package = "gpboost")
train <- agaricus.train
dtrain <- gpb.Dataset(train$data, label = train$label)
data_file <- tempfile(fileext = ".data")
gpb.Dataset.save(dtrain, data_file)
dtrain <- gpb.Dataset(data_file)
gpb.Dataset.set.categorical(dtrain, 1L:2L)

gpboost documentation built on July 14, 2021, 9:06 a.m.