Installation

Requirements

  • python 3

  • numpy >= 1.13

  • nltk >= 3.4

  • tqdm >= 4.30

  • checksumdir >= 1.1

  • pytorch >= 1.0.0 (optional, accelerating the calculation of some metrics)

  • transformers (optional, used for pretrained models)

We support Unix, Windows, and macOS.

Install from pip

You can simply get the latest stable version from pip using

pip install cotk

Install from source code

  • Clone the cotk repository

git clone https://github.com/thu-coai/cotk.git

  • Install cotk via pip

cd cotk
pip install -e .