コード例 #1
0
    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
コード例 #2
0
    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