import h5py

def load_shamnet4_traindata(fn):
    with h5py.File(fn, 'r') as hdf:
        smf_all_batches = hdf['smf_params'][...]
        shmf_all_batches = hdf['shmf_params'][...]
        scatter_all_batches = hdf['scatter_params'][...]
        smhm_all_batches = hdf['smhm_params'][...]
    return smf_all_batches, shmf_all_batches, scatter_all_batches, smhm_all_batches