Skip to content

Commit

Permalink
add arg introduction
Browse files Browse the repository at this point in the history
  • Loading branch information
KarhouTam committed Aug 24, 2024
1 parent ebdb9ad commit 00bddf6
Showing 1 changed file with 5 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -96,6 +96,10 @@ class ImageSemanticPartitioner(Partitioner):
Seed used for PCA dimensionality reduction. If not set, sample_seed is used.
gmm_seed: Optional[int]
Seed used for GMM clustering. If not set, sample_seed is used.
image_column_name: Optional[str]
The name of the image column in the dataset. If not set, the first image column
is used.
Examples
--------
Expand Down Expand Up @@ -498,4 +502,4 @@ def _check_variable_validation(self) -> None:
if not isinstance(self._pca_seed, int):
raise TypeError("The pca seed needs to be an integer.")
if not isinstance(self._gmm_seed, int):
raise TypeError("The gmm seed needs to be an integer.")
raise TypeError("The gmm seed needs to be an integer.")

0 comments on commit 00bddf6

Please sign in to comment.