12 lines
456 B
YAML
12 lines
456 B
YAML
# Спецификация окружения Python для MLflow Projects
|
|
# MLflow создаст виртуальное окружение по этому файлу
|
|
python: "3.10"
|
|
dependencies:
|
|
- mlflow>=2.11.0
|
|
- scikit-learn>=1.3.0
|
|
- pandas>=2.0.0
|
|
- numpy>=1.24.0
|
|
- matplotlib>=3.7.0
|
|
# Для GPU entry point установите torch отдельно:
|
|
# pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
|