Compare commits
2 Commits
b679e7d9c6
...
fb944fa73e
Author | SHA1 | Date | |
---|---|---|---|
fb944fa73e | |||
5fb057030a |
13
MLproject
Executable file
13
MLproject
Executable file
@ -0,0 +1,13 @@
|
||||
name: s424714
|
||||
|
||||
conda_env: environment.yml #ścieżka do pliku conda.yaml z definicją środowiska
|
||||
|
||||
entry_points:
|
||||
main:
|
||||
parameters:
|
||||
epochs: {type: int, default: 3}
|
||||
lr: {type: float, default: 0.000001}
|
||||
command: "python src/main.py --train --lr={lr} -e={epochs}"
|
||||
# command: "python src/train.py --train --lr={lr} -e={epochs}"
|
||||
validate:
|
||||
command: "python src/main.py --train"
|
@ -28,7 +28,7 @@ parser.add_argument("--num_epochs", "--epochs", "-e", type=int, default=3)
|
||||
|
||||
# mlflow stuff
|
||||
mlflow.set_tracking_uri("http://localhost:5000")
|
||||
mlflow.set_experiment("s424714")
|
||||
# mlflow.set_experiment("s424714")
|
||||
|
||||
|
||||
def main():
|
||||
|
Loading…
Reference in New Issue
Block a user