add conda env
Some checks failed
s464980-training/pipeline/head There was a failure building this commit

This commit is contained in:
Sheaza 2024-05-14 23:48:41 +02:00
parent 8ba3d2b97c
commit 1b646836f8
2 changed files with 22 additions and 0 deletions

9
MLproject Normal file
View 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
View File

@ -0,0 +1,13 @@
name: s464980
channels:
- defaults
dependencies:
- python=3.11
- pip
- pip:
- mlflow
- tensorflow
- pandas
- scikit-learn
- numpy
- matplotlib