pytorch_tabular icon indicating copy to clipboard operation
pytorch_tabular copied to clipboard

ValueError: y contains previously unseen labels

Open unolife opened this issue 1 year ago • 1 comments

Describe the bug I wanna do binary classification, but I got this error ValueError: y contains previously unseen labels: [-1.671504692016799]

data_config = DataConfig( target=[target_column], continuous_cols=continuous_cols, categorical_cols=categorical_cols, ) trainer_config = TrainerConfig( auto_lr_find=True, batch_size=1024, max_epochs=100, ) optimizer_config = OptimizerConfig() experiment_config = ExperimentConfig(project_name=seg_name)

model_config = TabNetModelConfig( task='classification', metrics=['accuracy', "f1_score"], metrics_params=[dict(task="multiclass", num_classes=2), dict(task="multiclass", num_classes=2)] )

model = TabularModel( model_config=model_config, data_config=data_config, optimizer_config=optimizer_config, trainer_config=trainer_config, experiment_config=experiment_config, ) model.fit(train=train, validation=valid) print(model.evaluate(test))

target have values 0 or 1, how can I fix?

To Reproduce

Expected behavior

Screenshots

Desktop (please complete the following information):

Smartphone (please complete the following information):

Additional context

unolife avatar Jan 24 '24 13:01 unolife

Can you check if the target column is part of the continuous or categorical features?

manujosephv avatar Jan 25 '24 03:01 manujosephv

Closing this, Do re-open if you feel the need to.

manujosephv avatar Jan 31 '24 23:01 manujosephv

~~same problem meet, target column is not part of continuous or categorical features~~

Never mind, Found an mistake that add target into continuous feature

HernandoR avatar Mar 06 '24 06:03 HernandoR