dfdf

Social-Evolution-and-Behavior · Apr 2, 2020 · ae9d3fd · ae9d3fd
1 parent bd6d01a
commit ae9d3fd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/antrax/classifier.py b/antrax/classifier.py
@@ -23,8 +23,8 @@
 from .models import new_model
 
 
-AMBIG_CLASSES = ['Unknown', '??', 'MultiAnt','Multi','multi']
-MULTI_CLASSES = ['MultiAnt','Multi','multi']
+AMBIG_CLASSES = ['Unknown', '??', 'MultiAnt', 'Multi', 'multi']
+MULTI_CLASSES = ['MultiAnt', 'Multi', 'multi']
 NONANT_CLASSES = ['NoAnt', 'FOOD', 'Larva']
 
 

diff --git a/antrax/cli.py b/antrax/cli.py
@@ -225,7 +225,7 @@ def solve(explist, *, glist: parse_movlist=None, clist: parse_movlist=None, mcr=
         Q.stop_workers()
 
 
-def train(classdir,  *, name='classifier', scratch=False, ne=5, unknown_weight: to_float=None, multi_weight: to_float=None,
+def train(classdir,  *, name='classifier', scratch=False, ne=5, unknown_weight=20, multi_weight=0.1,
           target_size: to_int=None, crop_size: to_int=None, hsymmetry=False, hpc=False, hpc_options: parse_hpc_options={}):