run_analysis.py

""" Script to run analysis on a single epoch"""
import argparse
import numpy as np
from comap_wrapper import stage_coMAP

import logging
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

def main():
    # Load file paths from command line arguments
    parser = argparse.ArgumentParser()
    parser.add_argument("--epoch_data_path", type=str, required=True)
    parser.add_argument("--model_info_path", type=str, required=True)
    parser.add_argument("--rnn_weights_path", type=str, required=True)
    parser.add_argument("--secondary_model_path", type=str, required=True)
    parser.add_argument("--refractory_period", type=int, required=True)
    args = parser.parse_args()

    # Load epoch data
    epoch_data = np.load(args.epoch_data_path)

    # Run analysis
    predictions, y_bin = stage_coMAP(epoch_data, args.model_info_path, args.rnn_weights_path, args.secondary_model_path, args.refractory_period)

    # Save predictions and y_bin
    np.save(args.predictions_path, predictions)
    np.save(args.y_bin_path, y_bin)

if __name__ == "__main__":
    main()