Example #1
0
    def _sample(self, graph):
        sampled_graph = GraphLearnSampler.transform_single(self, graph)

        if self.num_scores > self.MIN_SCORES_2_PREDICT:
            if self.num_scores - self.num_scores_fitted > self.MIN_BATCH_SIZE:
                self.create_features()
                self.cip_fit()

                print("FITTED ", self.num_scores_fitted)

        return sampled_graph