add scared fileobserver
Some checks failed
s440058-training/pipeline/head There was a failure building this commit
Some checks failed
s440058-training/pipeline/head There was a failure building this commit
This commit is contained in:
parent
46c50afa8c
commit
6731c7345f
@ -7,7 +7,6 @@ RUN pip3 install kaggle && pip3 install pandas && pip3 install scikit-learn && p
|
|||||||
RUN apt install -y curl
|
RUN apt install -y curl
|
||||||
RUN pip3 install --user wget
|
RUN pip3 install --user wget
|
||||||
RUN pip3 install torch torchvision torchaudio
|
RUN pip3 install torch torchvision torchaudio
|
||||||
RUN pip3 install sacred && pip3 install GitPython && pip3 install pymongo
|
|
||||||
|
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
|
|
||||||
|
@ -33,7 +33,7 @@ class LogisticRegressionModel(torch.nn.Module):
|
|||||||
return self.sigmoid(out)
|
return self.sigmoid(out)
|
||||||
|
|
||||||
@ex.capture
|
@ex.capture
|
||||||
def readAndtrain(num_epochs, batch_size, _run):
|
def script(num_epochs, batch_size, _run):
|
||||||
results = pd.read_csv('diabetes2.csv')
|
results = pd.read_csv('diabetes2.csv')
|
||||||
|
|
||||||
results.dropna()
|
results.dropna()
|
||||||
@ -91,8 +91,8 @@ def readAndtrain(num_epochs, batch_size, _run):
|
|||||||
_run.log_scalar("rmse", rmse)
|
_run.log_scalar("rmse", rmse)
|
||||||
|
|
||||||
@ex.automain
|
@ex.automain
|
||||||
def my_main(epochs, batch_size, _run):
|
def my_main(num_epochs, batch_size, _run):
|
||||||
readAndtrain()
|
script()
|
||||||
|
|
||||||
ex.run()
|
ex.run()
|
||||||
ex.add_artifact('diabetes.pth')
|
ex.add_artifact('diabetes.pth')
|
Loading…
Reference in New Issue
Block a user