update

Signed-off-by: Faakhir30 <zahidfaakhir@gmail.com>

Author: Faakhir30

pytorch_forecasting/models/temporal_fusion_transformer/_tft_pkg_v2.py

@@ -59,20 +59,17 @@ def get_test_train_params(cls):
                 hidden_size=16,
                 attention_head_size=4,
             ),
-            # non-point nn loss: mean/var head via free output_size
             dict(
                 loss=nn.GaussianNLLLoss(),
                 output_size=2,
                 hidden_size=16,
                 attention_head_size=2,
             ),
-            # class nn loss: logits head + discrete class_label target fixture
             dict(
                 loss=nn.CrossEntropyLoss(),
                 output_size=2,
                 hidden_size=16,
                 attention_head_size=2,
-                datamodule_cfg=dict(target="class_label"),
             ),
             dict(datamodule_cfg=dict(max_encoder_length=5, max_prediction_length=3)),
             dict(