def get_batch(self, tokenized_samples, labels): e = EasyEnsemble(random_state=0, n_subsets=1) e.fit(tokenized_samples, labels) X_resampled, y_resampled = e.sample(tokenized_samples, labels) X = X_resampled[0] y = y_resampled[0] targets = np.zeros(shape=(len(X), self._num_labels)) samples = np.zeros(shape=(len(X), self._max_document_length)) for sample_ix, sample in enumerate(X): label = y[sample_ix] targets[sample_ix, label] = 1 samples[sample_ix, :sample.shape[0]] = \ sample[:self._max_document_length] return samples, targets