From 6731c7345f28ae751b677884778a9d72fdff0928 Mon Sep 17 00:00:00 2001 From: piotr6789 Date: Mon, 24 May 2021 15:19:11 +0200 Subject: [PATCH] add scared fileobserver --- Dockerfile | 1 - scared-example-file.py | 6 +++--- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/Dockerfile b/Dockerfile index 0fe03a7..5938023 100644 --- a/Dockerfile +++ b/Dockerfile @@ -7,7 +7,6 @@ RUN pip3 install kaggle && pip3 install pandas && pip3 install scikit-learn && p RUN apt install -y curl RUN pip3 install --user wget RUN pip3 install torch torchvision torchaudio -RUN pip3 install sacred && pip3 install GitPython && pip3 install pymongo WORKDIR /app diff --git a/scared-example-file.py b/scared-example-file.py index 8a69446..f46743d 100644 --- a/scared-example-file.py +++ b/scared-example-file.py @@ -33,7 +33,7 @@ class LogisticRegressionModel(torch.nn.Module): return self.sigmoid(out) @ex.capture -def readAndtrain(num_epochs, batch_size, _run): +def script(num_epochs, batch_size, _run): results = pd.read_csv('diabetes2.csv') results.dropna() @@ -91,8 +91,8 @@ def readAndtrain(num_epochs, batch_size, _run): _run.log_scalar("rmse", rmse) @ex.automain -def my_main(epochs, batch_size, _run): - readAndtrain() +def my_main(num_epochs, batch_size, _run): + script() ex.run() ex.add_artifact('diabetes.pth') \ No newline at end of file