diff --git a/mongoObserver.py b/mongoObserver.py index d9313a3..dd5f52e 100644 --- a/mongoObserver.py +++ b/mongoObserver.py @@ -45,7 +45,7 @@ def readAndtrain(epochs, batch_size, _run): input_dim = 11 output_dim = 1 - _run.info("Batch", str(batch_size) + " epoch: " + str(epochs)) + _run.info("Batch", str(batch_size)) _run.info("epoch", str(epochs)) model = LogisticRegressionModel(input_dim, output_dim)