Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
nk-ruiying authored Aug 13, 2024
1 parent 56f7821 commit 202f085
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@
### Installation
Create a conda environment with pytorch and scikit-learn :
```
conda create --name teast_env python=3.8
source activate teast_env
conda create --name tcom_env python=3.8
source activate tcom_env
conda install --file requirements.txt -c pytorch
```

Expand All @@ -18,9 +18,9 @@ python process_gdelt.py

This will create the files required to compute the filtered metrics.

### Reproducing results of TeAST
### Reproducing results of TCompoundE

In order to reproduce the results of TeAST on the four datasets in the paper, run the following commands
In order to reproduce the results of TCompoundE on the four datasets in the paper, run the following commands

```
python learner.py --dataset ICEWS14 --emb_reg 0.01 --time_reg 0.01 --learning_rate 0.01 --rank 6000 --batch_size 4000 --max_epochs 400
Expand Down

0 comments on commit 202f085

Please sign in to comment.