fix get_auto_embedding_dim function

datawhalechina · Jan 2, 2025 · 466ab90 · 466ab90
1 parent 38340dd
commit 466ab90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_rechub/utils/data.py b/torch_rechub/utils/data.py
@@ -92,7 +92,7 @@ def get_auto_embedding_dim(num_classes):
     Returns:
         the dim of embedding vector
     """
-    return int(np.floor(6 * np.pow(num_classes, 0.26)))
+    return int(np.floor(6 * np.pow(num_classes, 0.25)))
 
 
 def get_loss_func(task_type="classification"):