exp_generators.py 文件源码

def load_mat_file(name, path, matname, load_zeros = False, prop_valid_set = .1, prop_test_set=0):

    x = scipy.io.loadmat(path + name)[matname]


    if sp.issparse(x): 
        if not load_zeros:
            idxs = x.nonzero()

            indexes = np.array(zip(idxs[0], np.zeros_like(idxs[0]), idxs[1]))
            np.random.shuffle(indexes)

            nb = indexes.shape[0]
            i_valid = int(nb - nb*prop_valid_set - nb * prop_test_set)
            i_test = i_valid + int( nb*prop_valid_set)

            train = Triplets_set(indexes[:i_valid,:], np.ones(i_valid))
            valid = Triplets_set(indexes[i_valid:i_test,:], np.ones(i_test - i_valid))
            test = Triplets_set(indexes[i_test:,:], np.ones(nb - i_test))


    return Experiment(name,train, valid, test, positives_only = True, compute_ranking_scores = True)