added enviroment yml file

This commit is contained in:
Maciej Tyczynski 2023-06-09 16:56:36 +02:00
parent 3ee6135028
commit 340e270dc8
2 changed files with 13 additions and 2 deletions

12
ium_env.yml Normal file
View File

@ -0,0 +1,12 @@
name: ium_env
channels:
- defaults
dependencies:
- python=3.10
- pytorch
- pandas
- requests
- scikit-learn
- seaborn
- mlflow
prefix: C:\Users\macty\.conda\envs\ium_env

View File

@ -1,11 +1,10 @@
import pandas as pd
import sklearn.model_selection
from datasets import load_dataset
import mlflow
import mlflow.sklearn
import numpy as np
import logging
from sklearn.metrics import mean_squared_error, mean_absolute_error, r2_score
logging.basicConfig(level=logging.WARN)
logger = logging.getLogger(__name__)