V-WSD

disambiguate.py

Use this script to disambiguate target words in context.

Installation:

Install pytorch following instructions from the official website.
Install dependencies bash consec/setup.sh
Download the ConSec "consec_wngt_best.ckpt" checkpoint from here
Move "consec_wngt_best.ckpt" to consec/checkpoints
Run disambigute.py in the following way PYTHONPATH=$(PWD)/consec python3 disambiguate.py --data_path data/trial.data.txt --output consec.tsv where --data_path is the SE23 dataset path and --output is the path where you would like to save the output

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
consec		consec
consec_train_output		consec_train_output
greg		greg
mlp_on_embeddings		mlp_on_embeddings
test.data.v1.1.gold		test.data.v1.1.gold
tests		tests
v1.translations/prediction		v1.translations/prediction
v2.lang-specific/prediction		v2.lang-specific/prediction
v3.multilingual		v3.multilingual
.gitignore		.gitignore
Mike's Notes.md		Mike's Notes.md
NOTES.md		NOTES.md
README.md		README.md
README.txt		README.txt
babelnet_baseline.py		babelnet_baseline.py
babelnet_baseline_plus_image_gen.py		babelnet_baseline_plus_image_gen.py
babelnet_baseline_plus_text_image.py		babelnet_baseline_plus_text_image.py
babelnet_conf.yml		babelnet_conf.yml
compute_stats.py		compute_stats.py
compute_stats_amuse.py		compute_stats_amuse.py
define.py		define.py
disambiguate.py		disambiguate.py
enumerate_sense_keys.py		enumerate_sense_keys.py
fetch_images.py		fetch_images.py
finetune_augmented_clip.py		finetune_augmented_clip.py
finetune_clip.py		finetune_clip.py
finetune_clip_orig.py		finetune_clip_orig.py
finetune_clip_small.py		finetune_clip_small.py
greg.txt		greg.txt
greg_augmented_create.txt		greg_augmented_create.txt
greg_augmented_create_fa.txt		greg_augmented_create_fa.txt
greg_augmented_create_it.txt		greg_augmented_create_it.txt
greg_augmented_create_wikt.txt		greg_augmented_create_wikt.txt
greg_augmented_data.txt		greg_augmented_data.txt
greg_augmented_data_all.txt		greg_augmented_data_all.txt
greg_augmented_data_all_hyper.txt		greg_augmented_data_all_hyper.txt
greg_augmented_data_hyper.txt		greg_augmented_data_hyper.txt
greg_augmented_stats.py		greg_augmented_stats.py
greg_search.100.log		greg_search.100.log
greg_search.500.log		greg_search.500.log
greg_search.json		greg_search.json
greg_search.log		greg_search.log
greg_search.py		greg_search.py
greg_simple.py		greg_simple.py
greg_simple_multilingual.py		greg_simple_multilingual.py
greg_simple_v2.py		greg_simple_v2.py
greg_sort.py		greg_sort.py
greg_v1.py		greg_v1.py
greg_v2.py		greg_v2.py
greg_v3.py		greg_v3.py
image_captioning_baseline.py		image_captioning_baseline.py
image_generation_baseline.py		image_generation_baseline.py
imagenet-synsets.json		imagenet-synsets.json
imagenet-synsets.txt		imagenet-synsets.txt
java		java
my_favorite_image.jpg		my_favorite_image.jpg
prediction.today.txt		prediction.today.txt
prediction.txt		prediction.txt
prediction.txt.zip		prediction.txt.zip
requirements.txt		requirements.txt
score.sh		score.sh
test.data.v1.1.gold.zip		test.data.v1.1.gold.zip
text_image_baseline.py		text_image_baseline.py
text_wsd_to_zero-shot_image_classification.py		text_wsd_to_zero-shot_image_classification.py
tmp.txt		tmp.txt
utils.py		utils.py
visual_question_answering_baseline.py		visual_question_answering_baseline.py

UAlberta-NLP/v-wsd

Folders and files

Latest commit

History

Repository files navigation

V-WSD

disambiguate.py

About

Resources

Stars

Watchers