Skip to content
Snippets Groups Projects

Resolve "Add some tests"

Merged Leo Bouscarrat requested to merge 19-add-some-tests into master
2 unresolved threads
2 files
+ 72
0
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -7,6 +7,20 @@ class ModelParameters(object):
@@ -7,6 +7,20 @@ class ModelParameters(object):
def __init__(self, extracted_forest_size, normalize_D, subsets_used,
def __init__(self, extracted_forest_size, normalize_D, subsets_used,
normalize_weights, seed, hyperparameters, extraction_strategy):
normalize_weights, seed, hyperparameters, extraction_strategy):
 
"""Init of ModelParameters.
 
 
Args:
 
extracted_forest_size (list): list of all the extracted forest
 
size.
 
normalize_D (bool): true normalize the distribution, false no
 
subsets_used (list): which dataset use for randomForest and for OMP
 
'train', 'dev' or 'train+dev' and combination of two of this.
 
normalize_weights (bool): if we normalize the weights or no.
 
seed (int): the seed used for the randomization.
 
hyperparameters (dict): dict of the hyperparameters of RandomForest
 
in scikit-learn.
 
extraction_strategy (str): either 'none', 'random', 'omp'
 
"""
self._extracted_forest_size = extracted_forest_size
self._extracted_forest_size = extracted_forest_size
self._normalize_D = normalize_D
self._normalize_D = normalize_D
self._subsets_used = subsets_used
self._subsets_used = subsets_used
Loading