TabNet Multi-Task Regressor

TabNet model class and training logic.

pytorch_tabnet.tab_model.MultiTabNetRegressor

alias of TabNetRegressor

Example

from pytorch_tabnet.tab_model import MultiTabNetRegressor
import numpy as np
X = np.random.rand(100, 10)
y = np.random.rand(100, 3)  # 3 regression targets
reg = MultiTabNetRegressor()
reg.fit(X_train=X, y_train=y)
preds = reg.predict(X)