Skip to content

Installation

This section explains how to install pytorch-widedeep.

For the latest stable release, execute:

pip install pytorch-widedeep

For the bleeding-edge version, execute:

pip install git+https://github.com/jrzaurin/pytorch-widedeep.git

For developer install

# Clone the repository
git clone https://github.com/jrzaurin/pytorch-widedeep
cd pytorch-widedeep

# Install in dev mode
pip install -e .

Dependencies

  • pandas>=1.3.5
  • numpy>=1.21.6
  • scipy>=1.7.3,<=1.12.0
  • scikit-learn>=1.0.2
  • gensim
  • spacy
  • opencv-contrib-python
  • imutils
  • tqdm
  • torch >= 2.0.0
  • torchvision >= 0.15.0
  • einops
  • wrapt
  • torchmetrics
  • pyarrow
  • fastparquet>=0.8.1
  • transformers
  • sentence-transformers
  • sentencepiece