epsilon

Load the Epsilon dataset.

This dataset is best suited for binary classification.

The training dataset contains 400000 objects. Each object is described by 2001 columns. The first column contains the label value, all other columns contain numerical features.

The test dataset contains 100000 objects. The structure is identical to the training dataset.

Method call format

epsilon()

Type of return value

A two pandas.DataFrame tuple (for train and test datasets).

Usage examples

from catboost.datasets import epsilon
epsilon_train, epsilon_test = epsilon()