def get_dataset_id(data_id): options_data = dataset.get_yaml_options() options = dict_to_object(options_data[data_id]) filename = options.filename + "." + options.type filepath = dataset.learning_data_dir() + os.sep + filename if os.path.exists(filepath) and os.path.isfile(filepath): data = dataset.CostmapDataset(filename) numtrain = data.train_inputs.shape[0] numtest = data.test_inputs.shape[0] numdatasets = numtrain + numtest assert options.numdatasets == numdatasets assert options.xsize == data.train_targets.shape[1] assert options.ysize == data.train_targets.shape[2] assert options.xsize == data.train_inputs.shape[1] assert options.ysize == data.train_inputs.shape[2] assert options.xsize == data.test_targets.shape[1] assert options.ysize == data.test_targets.shape[2] assert options.xsize == data.test_inputs.shape[1] assert options.ysize == data.test_inputs.shape[2] return data else: datasets, workspaces = random_environments(options) dataset.write_dictionary_to_file(datasets, filename) dataset.write_dictionary_to_file( workspaces, options.workspaces + "." + options.type) return get_dataset_id(data_id)
def get_options(self): """ Load dataset options from file or option parser """ if self._use_parser: parser = self.environment_parser() (options, args) = parser.parse_args() return options else: options = dataset.get_yaml_options()[self._dataset_id] return dict_to_object(options)
dest='seed', help='Random number seed. -ve values\ mean random seed') return parser def get_options(self): """ Load dataset options from file or option parser """ if self._use_parser: parser = self.environment_parser() (options, args) = parser.parse_args() return options else: options = dataset.get_yaml_options()[self._dataset_id] return dict_to_object(options) def remove_file_if_exists(file): if os.path.exists(file): os.remove(file) if __name__ == '__main__': parser = RandomEnvironmentOptions() options = parser.get_options() dataset_paramerters = dict_to_object( dataset.get_yaml_options()[options.dataset_id]) remove_file_if_exists("data/" + dataset_paramerters.filename + ".hdf5") remove_file_if_exists("data/" + dataset_paramerters.workspaces + ".hdf5") get_dataset_id(options.dataset_id)