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
