def save_models_to_os()

in src/save_models.py [0:0]

• 9 lines of code
• 5 McCabe index (conditional complexity)

def save_models_to_os(models, dirname):
    """
    Writes a list of models to a file
    """
    # Setup path string
    base_dir = os.path.dirname(__file__)[:-4]  # we use -4 to take off the src/ from the end to go back a directory
    results_dir = os.path.join(base_dir, f'{dirname}/Models/')

    # Create the directory, if needed
    if not os.path.isdir(results_dir):
        print(f'Making directory: {results_dir}')
        os.makedirs(results_dir)

    # NOTE: named_models is no longer 1-indexed but the model in position 0 is called 'model_1'
    named_models = [(models[i], f'model_{i}') for i in range(1, len(models))]

    # Write each model to its own file within directory `dirname`
    for model, model_name in named_models:
        dump(model, results_dir + model_name)  # Write model to file