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
