Closed omkarmutreja12 closed 2 years ago
There are 3 output classes and it's a multi-class classification problem Shapes of train, valid and test also look fine, could you please help me out with the solution?
It looks like you might have unknown categorical values in one of your columns.
Closing this, feel free to reopen with more information.
####### Code ###########
clf = TabNetClassifier( n_d=64, n_a=64, n_steps=2, gamma=1.5, n_independent=1, n_shared=1, cat_idxs=cat_idx, cat_dims=cat_dims,
cat_emb_dim=1,
)
clf.fit( X_train=X_train, y_train=y_train, eval_set=[(X_train, y_train), (X_valid, y_valid)], eval_name=['train', 'valid'], max_epochs=max_epochs , patience=10, batch_size=8, virtual_batch_size=8,
num_workers=0,
)
########### Error ###########3
IndexError Traceback (most recent call last)