update sklearn

Author: simonblauth

pytorch_fob/tasks/tabular/model.py

@@ -1,5 +1,5 @@
 import torch
-from sklearn.metrics import r2_score, mean_squared_error
+from sklearn.metrics import r2_score, root_mean_squared_error
 from rtdl_revisiting_models import FTTransformer, _CLSEmbedding, LinearEmbeddings, CategoricalEmbeddings
 from pytorch_fob.engine.configs import TaskConfig
 from pytorch_fob.engine.parameter_groups import GroupedModel, ParameterGroup, group_named_parameters
@@ -80,7 +80,7 @@ def compute_and_log_metrics(self, preds: torch.Tensor, targets: torch.Tensor, st
         preds = preds.detach().cpu().float().numpy()
         targets = targets.detach().cpu().float().numpy()
         metrics = {
-            "rmse": mean_squared_error(targets, preds, squared=False),
+            "rmse": root_mean_squared_error(targets, preds),
             "r2_score": r2_score(targets, preds)
         }
         for k, v in metrics.items():

requirements.txt

@@ -10,7 +10,7 @@ tqdm~=4.66.1
 wget~=3.2
 deepspeed~=0.12.6
 rtdl_revisiting_models~=0.0.2
-scikit-learn~=1.3.2
+scikit-learn~=1.5.0
 transformers~=4.38.0
 tokenizers~=0.15.0
 sentencepiece~=0.2.0