Commit b3616f27 by simonabottani

ADD RB Rep K fold

parent 32a12152
......@@ -507,7 +507,7 @@ class VB_LearningCurveRepHoldOut_DualSVM(base.MLWorkflow):
class RB_RepKFold_DualSVM(base.MLWorkflow):
def __init__(self, caps_directory, subjects_visits_tsv, diagnoses_tsv, group_id, image_type, atlas,
output_dir, pvc=None, n_threads=15, n_iterations=100, test_size=0.3,
output_dir, pvc=None, n_threads=15, n_iterations=100, test_size=0.3,n_folds = 10,
grid_search_folds=10, balanced=True, c_range=np.logspace(-6, 2, 17), splits_indices=None):
self._output_dir = output_dir
self._n_threads = n_threads
......@@ -516,6 +516,7 @@ class RB_RepKFold_DualSVM(base.MLWorkflow):
self._grid_search_folds = grid_search_folds
self._balanced = balanced
self._c_range = c_range
self._n_folds = n_folds
self._splits_indices = splits_indices
self._input = input.CAPSRegionBasedInput(caps_directory, subjects_visits_tsv, diagnoses_tsv, group_id,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment