Pool initialization

Loading data from a file

A list of possible methods to load the dataset from a file is given in the table below.
MethodUsage exampleInformation
Use default column descriptions
Pool(dataset_desc_file)

If the column descriptions file is omitted, it is assumed that the first column in the file with the dataset description defines the label value, and the other columns are the values of numerical features.

Use specified column descriptions
Pool(dataset_desc_file, column_description=cd_file)
If specified, the cd_file should contain the column descriptions.

Loading data from array_like structures

A list of possible methods to load the dataset from array-like structures is given in the table below.
MethodUsage exampleInformation
Use numerical features only
df = pd.read_table(TRAIN_FILE)
Pool(data=df.iloc[:, 1:].values, label=df.iloc[:, 0].values)
It is assumed that all features are numerical, since cat_features are not defined.
Use both numerical and categorical features
df = pd.read_table(TRAIN_FILE)
Pool(data=df.iloc[:, 1:].values, label=df.iloc[:, 0].values, cat_features=[1,2,3])
It is assumed that the list of feature indices specified in the cat_features parameter correspond to categorical features. All other features are considered numerical.