s440058/ium_440058
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add scared fileobserver
Some checks failed
s440058-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
piotr6789 2021-05-24 15:19:11 +02:00
parent 46c50afa8c
commit 6731c7345f
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Dockerfile
View File

@ -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

6
scared-example-file.py
View File

@ -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')