Method pld_train_preprocess

python/graphvite/dataset.py:505–508

(self, graph_file, train_file)

        )

    def pld_train_preprocess(self, graph_file, train_file):
        valid_file = train_file[:train_file.rfind("pld_train.txt")] + "pld_valid.txt"
        test_file = train_file[:train_file.rfind("pld_train.txt")] + "pld_test.txt"
        self.link_prediction_split(graph_file, [train_file, valid_file, test_file], portions=[10000, 1, 1])

    def pld_valid_preprocess(self, graph_file, valid_file):
        train_file = valid_file[:valid_file.rfind("pld_valid.txt")] + "pld_train.txt"