Remove class limit

extract_features.py

@@ -19,7 +19,7 @@ from tqdm import tqdm
 
 # Set defaults.
 seq_length = 40
-class_limit = 2  # integer, number of classes to extract
+class_limit = None  # Number of classes to extract. Can be 1-101 or None for all.
 
 # Get the dataset.
 data = DataSet(seq_length=seq_length, class_limit=class_limit)

train.py

@@ -85,9 +85,9 @@ def train(data_type, seq_length, model, saved_model=None,
 def main():
     """These are the main training settings. Set each before running
     this file."""
-    model = 'conv_3d'  # see `models.py` for more
+    model = 'lstm'  # see `models.py` for more
     saved_model = None  # None or weights file
-    class_limit = 2  # int, can be 1-101 or None
+    class_limit = Non  # int, can be 1-101 or None
     seq_length = 40
     load_to_memory = True  # pre-load the sequences into memory