Skip to content

Commit fdc4b32

Browse files
committed
update args
1 parent 28f894b commit fdc4b32

File tree

1 file changed

+0
-9
lines changed

1 file changed

+0
-9
lines changed

autointent/modules/scoring/_sklearn/sklearn_scorer.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -47,28 +47,19 @@ def __init__(
4747
embedder_max_length: int | None = None,
4848
embedder_device: str = "cpu",
4949
embedder_use_cache: bool = True,
50-
cv: int = 3,
5150
clf_args: dict[str, Any] | None = None,
52-
n_jobs: int = -1,
53-
seed: int = 0,
5451
) -> None:
5552
"""
5653
Initialize the SklearnScorer.
5754
5855
:param embedder_name: Name of the embedder model.
5956
:param clf_name: Name of the sklearn classifier to use.
60-
:param cv: Number of cross-validation folds, defaults to 3.
6157
:param clf_args: dictionary with the chosen sklearn classifier arguments, defaults to {}.
62-
:param n_jobs: Number of parallel jobs for cross-validation, defaults to -1 (all CPUs).
63-
:param seed: Random seed for reproducibility, defaults to 0.
6458
:param embedder_batch_size: Batch size for embedding generation, defaults to 32.
6559
:param embedder_max_length: Maximum sequence length for embedding, or None for default.
6660
:param embedder_device: Device to run operations on, e.g., "cpu" or "cuda".
6761
:param embedder_use_cache: Flag indicating whether to cache intermediate embeddings.
6862
"""
69-
self.cv = cv
70-
self.n_jobs = n_jobs
71-
self.seed = seed
7263
self.embedder_name = embedder_name
7364
self.clf_name = clf_name
7465
self.clf_args = clf_args or {}

0 commit comments

Comments
 (0)