add conda env
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
8ba3d2b97c
commit
1b646836f8
9
MLproject
Normal file
9
MLproject
Normal file
@ -0,0 +1,9 @@
|
|||||||
|
name: s464980
|
||||||
|
|
||||||
|
conda_env: conda.yaml
|
||||||
|
|
||||||
|
entry_points:
|
||||||
|
optimal_parameters:
|
||||||
|
parameters:
|
||||||
|
epochs: { type: int, default: 20 }
|
||||||
|
command: 'python train.py {epochs}'
|
13
conda.yaml
Normal file
13
conda.yaml
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
name: s464980
|
||||||
|
channels:
|
||||||
|
- defaults
|
||||||
|
dependencies:
|
||||||
|
- python=3.11
|
||||||
|
- pip
|
||||||
|
- pip:
|
||||||
|
- mlflow
|
||||||
|
- tensorflow
|
||||||
|
- pandas
|
||||||
|
- scikit-learn
|
||||||
|
- numpy
|
||||||
|
- matplotlib
|
Loading…
Reference in New Issue
Block a user