diff --git a/comet.py b/comet.py index 51efd685e97fb61e0d26682893746018ee1e138c..e2f062b91fe5ec0d9bd6636d26a83414c51dfdfe 100644 --- a/comet.py +++ b/comet.py @@ -16,9 +16,6 @@ class MulticlassificationConfusionMatrix(tf.keras.callbacks.Callback): def on_train_begin(self, logs=None): pred = self.model.predict(x=self.data_x, batch_size=self.batch_size) - print("DEBUGGING!") - print(pred.shape) - print(self.data_label.shape) self.comet_experiment.log_confusion_matrix(y_true=self.data_label, y_predicted=pred, labels=self.labels,