File tree Expand file tree Collapse file tree 1 file changed +0
-9
lines changed
autointent/modules/scoring/_sklearn Expand file tree Collapse file tree 1 file changed +0
-9
lines changed Original file line number Diff line number Diff 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 {}
You can’t perform that action at this time.
0 commit comments