text_plain_output_1.png
from pytorch_tabnet.tab_model import TabNetClassifier

from pytorch_tabnet.tab_model import TabNetClassifier
clf = TabNetClassifier()
clf.fit(X_train, y_train, eval_set=[(X_test, y_test)], max_epochs=2)