diff --git a/environment.yml b/environment.yml
new file mode 100644
index 0000000..d7aa83a
--- /dev/null
+++ b/environment.yml
@@ -0,0 +1,137 @@
+name: s444018
+channels:
+  - scikit-learn
+  - pytorch
+  - conda-forge
+  - anaconda
+  - defaults
+dependencies:
+  - alembic=1.7.7=pyhd8ed1ab_0
+  - appdirs=1.4.4=pyh9f0ad1d_0
+  - asn1crypto=1.5.1=pyhd8ed1ab_0
+  - brotli=1.0.9=h5eb16cf_7
+  - brotli-bin=1.0.9=h5eb16cf_7
+  - brotlipy=0.7.0=py310h1961e1f_1004
+  - bzip2=1.0.8=h0d85af4_4
+  - ca-certificates=2022.5.18.1=h033912b_0
+  - certifi=2022.5.18.1=py310h2ec42d9_0
+  - cffi=1.15.0=py310hcc37b68_0
+  - charset-normalizer=2.0.12=pyhd8ed1ab_0
+  - click=8.1.3=py310h2ec42d9_0
+  - cloudpickle=2.1.0=pyhd8ed1ab_0
+  - colorama=0.4.4=pyh9f0ad1d_0
+  - configparser=5.2.0=pyhd8ed1ab_0
+  - cryptography=37.0.2=py310h52c3658_0
+  - cycler=0.11.0=pyhd8ed1ab_0
+  - databricks-cli=0.12.1=pyhd8ed1ab_0
+  - docker-py=5.0.3=py310h2ec42d9_2
+  - docker-pycreds=0.4.0=py_0
+  - entrypoints=0.4=pyhd8ed1ab_0
+  - flask=2.1.2=pyhd8ed1ab_1
+  - fonttools=4.33.3=py310h6c45266_0
+  - freetype=2.10.4=h4cff582_1
+  - giflib=5.2.1=hbcb3906_2
+  - gitdb=4.0.9=pyhd8ed1ab_0
+  - gitpython=3.1.27=pyhd8ed1ab_0
+  - greenlet=1.1.2=py310h9d931ec_2
+  - gunicorn=20.1.0=py310h2ec42d9_2
+  - idna=3.3=pyhd8ed1ab_0
+  - importlib-metadata=4.11.4=py310h2ec42d9_0
+  - importlib_resources=5.7.1=pyhd8ed1ab_1
+  - itsdangerous=2.1.2=pyhd8ed1ab_0
+  - jbig=2.1=h0d85af4_2003
+  - jinja2=3.1.2=pyhd8ed1ab_0
+  - joblib=1.1.0=pyhd8ed1ab_0
+  - jpeg=9e=h5eb16cf_1
+  - kaggle=1.5.12=pyhd8ed1ab_4
+  - kiwisolver=1.4.2=py310h6be76da_1
+  - lcms2=2.12=h577c468_0
+  - lerc=3.0=he49afe7_0
+  - libblas=3.9.0=14_osx64_mkl
+  - libbrotlicommon=1.0.9=h5eb16cf_7
+  - libbrotlidec=1.0.9=h5eb16cf_7
+  - libbrotlienc=1.0.9=h5eb16cf_7
+  - libcblas=3.9.0=14_osx64_mkl
+  - libcxx=14.0.3=hc203e6f_0
+  - libdeflate=1.10=h0d85af4_0
+  - libffi=3.4.2=h0d85af4_5
+  - libgfortran=5.0.0=9_3_0_h6c81a4c_23
+  - libgfortran5=9.3.0=h6c81a4c_23
+  - liblapack=3.9.0=14_osx64_mkl
+  - libopenblas=0.3.20=openmp_hb3cd9ec_0
+  - libpng=1.6.37=h7cec526_2
+  - libprotobuf=3.20.1=h2292cb8_0
+  - libtiff=4.3.0=h17f2ce3_3
+  - libwebp=1.2.2=h28dabe5_0
+  - libwebp-base=1.2.2=h0d85af4_1
+  - libxcb=1.13=h0d85af4_1004
+  - libzlib=1.2.11=h6c3fc93_1014
+  - llvm-openmp=14.0.3=ha654fa7_0
+  - lz4-c=1.9.3=he49afe7_1
+  - mako=1.2.0=pyhd8ed1ab_1
+  - markupsafe=2.1.1=py310h1961e1f_1
+  - matplotlib=3.5.2=py310h2ec42d9_0
+  - matplotlib-base=3.5.2=py310h4510841_0
+  - mkl=2022.1.0=h860c996_928
+  - mlflow=1.26.0=py310h73f00d3_0
+  - munkres=1.1.4=pyh9f0ad1d_0
+  - ncurses=6.3=h96cf925_1
+  - ninja=1.11.0=h1b54a9f_0
+  - numpy=1.22.3=py310he1f95b8_2
+  - openjpeg=2.4.0=h6e7aa92_1
+  - openssl=1.1.1o=hfe4f2af_0
+  - packaging=21.3=pyhd8ed1ab_0
+  - pandas=1.4.2=py310h514ec25_1
+  - pillow=9.1.1=py310hb3240ae_0
+  - pip=22.1.1=pyhd8ed1ab_0
+  - prometheus_client=0.14.1=pyhd8ed1ab_0
+  - prometheus_flask_exporter=0.20.1=pyhd8ed1ab_0
+  - protobuf=3.20.1=py310hd4537e4_0
+  - pthread-stubs=0.4=hc929b4f_1001
+  - pycparser=2.21=pyhd8ed1ab_0
+  - pyopenssl=22.0.0=pyhd8ed1ab_0
+  - pyparsing=3.0.9=pyhd8ed1ab_0
+  - pysocks=1.7.1=py310h2ec42d9_5
+  - python=3.10.4=h8b4d769_0_cpython
+  - python-dateutil=2.8.2=pyhd8ed1ab_0
+  - python-slugify=6.1.2=pyhd8ed1ab_0
+  - python_abi=3.10=2_cp310
+  - pytorch=1.11.0=cpu_py310h5ac1a50_1
+  - pytz=2022.1=pyhd8ed1ab_0
+  - pyyaml=6.0=py310h1961e1f_4
+  - querystring_parser=1.2.4=py_0
+  - readline=8.1=h05e3726_0
+  - requests=2.27.1=pyhd8ed1ab_0
+  - scikit-learn=1.1.1=py310hfc06b38_0
+  - scipy=1.8.1=py310h1f9c157_0
+  - setuptools=62.3.2=py310h2ec42d9_0
+  - six=1.16.0=pyh6c4a22f_0
+  - sleef=3.5.1=h6db0672_2
+  - smmap=3.0.5=pyh44b312d_0
+  - sqlalchemy=1.4.36=py310h6c45266_0
+  - sqlite=3.38.5=hd9f0692_0
+  - sqlparse=0.4.2=pyhd8ed1ab_0
+  - tabulate=0.8.9=pyhd8ed1ab_0
+  - tbb=2021.5.0=hbb4e6a2_1
+  - tenacity=8.0.1=pyhd8ed1ab_0
+  - text-unidecode=1.3=py_0
+  - threadpoolctl=3.1.0=pyh8a188c0_0
+  - tk=8.6.12=h5dbffcc_0
+  - tornado=6.1=py310h1961e1f_3
+  - tqdm=4.64.0=pyhd8ed1ab_0
+  - typing_extensions=4.2.0=pyha770c72_1
+  - tzdata=2022a=h191b570_0
+  - unicodedata2=14.0.0=py310h1961e1f_1
+  - unidecode=1.3.4=pyhd8ed1ab_0
+  - urllib3=1.26.9=pyhd8ed1ab_0
+  - websocket-client=1.3.2=pyhd8ed1ab_0
+  - werkzeug=2.1.2=pyhd8ed1ab_1
+  - wheel=0.37.1=pyhd8ed1ab_0
+  - xorg-libxau=1.0.9=h35c211d_0
+  - xorg-libxdmcp=1.1.3=h35c211d_0
+  - xz=5.2.5=haf1e3a3_1
+  - yaml=0.2.5=h0d85af4_2
+  - zipp=3.8.0=pyhd8ed1ab_0
+  - zlib=1.2.11=h6c3fc93_1014
+  - zstd=1.5.2=ha9df2e0_1
+prefix: /Users/szymonparafinski/opt/miniconda3/envs/s444018