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