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, 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 self._n_iterations = n_iterations self._test_size = test_size 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, image_type, atlas, pvc) self._validation = None self._algorithm = None
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, grid_search_folds=10, balanced=True, n_estimators_range=(100, 200, 400), max_depth_range=[None], min_samples_split_range=[2], max_features_range=('auto', 0.25, 0.5), splits_indices=None): self._output_dir = output_dir self._n_threads = n_threads self._n_iterations = n_iterations self._test_size = test_size self._grid_search_folds = grid_search_folds self._balanced = balanced self._n_estimators_range = n_estimators_range self._max_depth_range = max_depth_range self._min_samples_split_range = min_samples_split_range self._max_features_range = max_features_range self._splits_indices = splits_indices self._input = input.CAPSRegionBasedInput(caps_directory, subjects_visits_tsv, diagnoses_tsv, group_id, image_type, atlas, pvc) self._validation = None self._algorithm = None