Skip to content

Commit 9f28ce7

Browse files
authored
Fix for PCA scikit-learn benchmark (#94)
1 parent 0cf7cce commit 9f28ce7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sklearn_bench/pca.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def main():
5858
parser = argparse.ArgumentParser(description='scikit-learn PCA benchmark')
5959
parser.add_argument('--svd-solver', type=str, choices=['full'],
6060
default='full', help='SVD solver to use')
61-
parser.add_argument('--n-components', type=float, default=None,
61+
parser.add_argument('--n-components', type=int, default=None,
6262
help='The number of components to find')
6363
parser.add_argument('--whiten', action='store_true', default=False,
6464
help='Perform whitening')

0 commit comments

Comments
 (0)