Function load_txt

data_preprocessing/merge_data/merge_weight.py:8–18

(file_name)

root = '/YOUR/DATA/DIR'

def load_txt(file_name):
    with open(file_name) as f:
        data = f.readlines()
    n = int(data[0].split(" ")[0])
    m = int(data[0].split(" ")[1])
    weights = np.zeros((n, m))
    for i in range(n):
        w = data[i + 1].split(" ")
        for j in range(m):
            weights[i][j] = float(w[j])
    return weights

for cat in categories:
    with open("all_%s.txt" % cat) as f: