Temp class limit change

57f7935e · Ubuntu · ebbaebc6 · 57f7935e
Commit 57f7935e authored 7 years ago by Ubuntu
Hide whitespace changes
Inline Side-by-side

Showing

with 1 addition and 1 deletion
+1 -1
--- a/extract_features.py
+++ b/extract_features.py
@@ -19,7 +19,7 @@ from tqdm import tqdm

 # Set defaults.
 seq_length = 40
-class_limit = None  # integer, number of classes to extract
+class_limit = 2  # integer, number of classes to extract

 # Get the dataset.
 data = DataSet(seq_length=seq_length, class_limit=class_limit)