import numpy as np
import matplotlib.pyplot as plt

# each of these is a 62 x 1000 array of the chrombpnet predicted counts for 62 snps.
s6_alt_profiles = np.load("s6_alt_profiles.npy")
s24_alt_profiles = np.load("s24_alt_profiles.npy")
c1_alt_profiles = np.load("c1_alt_profiles.npy")
s6_ref_profiles = np.load("s6_ref_profiles.npy")
s24_ref_profiles = np.load("s24_ref_profiles.npy")
c1_ref_profiles = np.load("c1_ref_profiles.npy")

# the rsids are in order
rsid_file = "rsids.txt"

with open(rsid_file, "r") as f:
    rsids = f.readlines() # this is a list of the rsids in order corresponding to the rows of the profiles



print(s6_alt_profiles.shape, s24_alt_profiles.shape, c1_alt_profiles.shape, s6_ref_profiles.shape, s24_ref_profiles.shape, c1_ref_profiles.shape)