Method generate_math

python/graphvite/dataset.py:602–609

(self, save_file, num_triplet)

        self.generate_math(save_file, num_triplet=1000)

    def generate_math(self, save_file, num_triplet):
        with open(save_file, "w") as fout:
            for _ in range(num_triplet):
                i = int(np.random.rand() * len(self.OPERATORS))
                op, f = self.OPERATORS[i]
                x = int(np.random.rand() * self.NUM_ENTITY)
                y = int(np.random.rand() * self.NUM_RELATION) + 1
                fout.write("%d\t%s%d\t%d\n" % (x, op, y, f(x, y)))


class FB15k(Dataset):