diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 6cfca422..52dd13fe 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -33,7 +33,7 @@ repos: # - --remove-header - --license-filepath - LICENSE_HEADER - - --use-current-year + # - --use-current-year - repo: https://github.com/pre-commit/pre-commit-hooks rev: v5.0.0 hooks: diff --git a/mostlyai/engine/_tabular/argn.py b/mostlyai/engine/_tabular/argn.py index b26d8a89..f29a36a9 100644 --- a/mostlyai/engine/_tabular/argn.py +++ b/mostlyai/engine/_tabular/argn.py @@ -117,9 +117,9 @@ def _embedding_heuristic(id: str, model_size: ModelSizeOrUnits, dim_input: int) if isinstance(model_size, dict): return model_size[id] model_size_output_dim = dict( - S=max(10, int(2 * np.ceil(dim_input**0.15))), - M=max(10, int(3 * np.ceil(dim_input**0.25))), - L=max(10, int(4 * np.ceil(dim_input**0.33))), + S=int(2 * np.ceil(dim_input**0.15)), + M=int(3 * np.ceil(dim_input**0.25)), + L=int(4 * np.ceil(dim_input**0.33)), ) return min(dim_input, model_size_output_dim[model_size])