def _sample(self, graph):
sampled_graph = GraphLearnSampler._sample(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
