sacred fix 1
Some checks failed
s464980-training/pipeline/head There was a failure building this commit
Some checks failed
s464980-training/pipeline/head There was a failure building this commit
This commit is contained in:
parent
7a3e24a150
commit
343ebbf307
@ -54,7 +54,7 @@ class RegressionModel:
|
||||
|
||||
def save_model(self):
|
||||
self.model.save("model.keras")
|
||||
ex.add_artifact("model.keras")
|
||||
ex.add_artifact("model.keras", content_type="application/octet-stream")
|
||||
|
||||
|
||||
@ex.main
|
||||
|
Loading…
Reference in New Issue
Block a user