def generate_model_with_parameters(arguments): """Trains a model with the given parameters, saving results.""" param_str = gen_param_str(arguments) out_dir_name = arguments.model_directory + param_str + '/' model_training_util.mkdir_p(out_dir_name) # to set more fine-grained parameters about MIRT training, look at # the arguments at mirt/mirt_train_EM.py mirt_train_params = [ '-a', str(arguments.abilities), '-w', str(arguments.workers), '-n', str(arguments.num_epochs), '-f', arguments.model_directory + 'train.responses', '-o', out_dir_name, ] if arguments.time: mirt_train_params.append('--time') mirt_train_EM.run_programmatically(mirt_train_params)
def make_necessary_directories(arguments): """Ensure that output directories for the data we'll be writing exist.""" roc_dir = arguments.model_directory + 'rocs/' model_training_util.mkdir_p([roc_dir])