Skip to content

Commit

Permalink
Update classification.py
Browse files Browse the repository at this point in the history
  • Loading branch information
gcattan authored Feb 5, 2025
1 parent 0054930 commit f1fe307
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions pyriemann_qiskit/classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -867,7 +867,8 @@ class NearestConvexHull(BaseEstimator, ClassifierMixin, TransformerMixin):
The number of hulls used per class, when subsampling is "random".
n_samples_per_hull : int, default=15
Defines how many samples are used to build a hull. -1 will include
all samples per class.
all samples per class. If subsampling is "full", this
parameter is defaulted to -1.
subsampling : {"min", "random", "full"}, default="min"
Subsampling strategy of training set to estimate distance to hulls.
"min" estimates hull using the n_samples_per_hull closest matrices.
Expand Down Expand Up @@ -903,7 +904,7 @@ def __init__(

if subsampling not in ["min", "random", "full"]:
raise ValueError(f"Unknown subsampling type {subsampling}.")

if subsampling == "full":
# From code perspective, "full" strategy is the same as min strategy
# without sorting
Expand Down

0 comments on commit f1fe307

Please sign in to comment.