Skip to content

Commit

Permalink
Updated README.md.
Browse files Browse the repository at this point in the history
  • Loading branch information
TimDettmers committed Nov 9, 2017
1 parent 074627b commit 1d99b28
Showing 1 changed file with 7 additions and 4 deletions.
11 changes: 7 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -97,11 +97,14 @@ lr_decay = learning_rate_decay
lr = learning_rate
label_smoothing = label_smoothing_epsilon
```
The parameters with the equal sign are equivalent and short-forms of each other.

To reproduce most of the results in the ConvE paper, you can use command below:

```
CUDA_VISIBLE_DEVICES=0 python main.py model ConvE input_drop 0.2 hidden_drop 0.3 \
feat_drop 0.2 lr 0.003 dataset DATASET_NAME
CUDA_VISIBLE_DEVICES=0 python main.py model ConvE input_drop 0.2 hidden_drop 0.3 \
feat_drop 0.2 lr 0.003 lr_decay 0.995 \
dataset DATASET_NAME
```
For the reverse model, you can run the provided file with the name of the dataset name and a threshold probability:

Expand All @@ -113,9 +116,9 @@ python reverse_rule.py WN18RR 0.9

To run it on a new datasets, copy your dataset folder into the data folder and make sure your dataset split files have the name `train.txt`, `valid.txt`, and `test.txt` which contain tab separated triples of a knowledge graph. Then execute `python wrangle_KG.py FOLDER_NAME`, afterwards, you can use the folder name of your dataset in the dataset parameter.

### Adding new models
### Adding your own model

A barebone model `MyModel` can be found in the `model.py` file. You can easily extend this to create your own link-prediction model.
You can easily write your own knowledge graph model by extending the barebone model `MyModel` that can be found in the `model.py` file.

### Quirks

Expand Down

0 comments on commit 1d99b28

Please sign in to comment.