docs: def main(model_folder):
      """Main part of the training script."""
      model_description_file = os.path.join(model_folder, "info.yml")
  
      # Read the model description file
      with open(model_description_file, 'r') as ymlfile:
          model_description = yaml.load(ymlfile)
  
      # Analyze model
      logging.info(model_description['model'])
      data = {}
      data['training'] = os.path.join(model_folder, "traindata.hdf5")
      data['testing'] = os.path.join(model_folder, "testdata.hdf5")
      data['validating'] = os.path.join(model_folder, "validdata.hdf5")
      train_model(model_folder)