Index

A

annotate() (srtk.entity_linking.dbpedia.DBpediaLinker method) (srtk.entity_linking.linker_base.LinkerBase method) (srtk.entity_linking.wikidata.WikidataLinker method)

annotate_rel() (srtk.entity_linking.wikidata.WikidataLinker method) annotate_tagme_wat() (srtk.entity_linking.wikidata.WikidataLinker method) avg_pool() (srtk.scorer.encoder.LitSentenceEncoder static method)

B

batch_forward() (srtk.scorer.encoder.LitSentenceEncoder method)

batch_score() (srtk.scorer.scorer.Scorer method) beam_search_path() (srtk.retrieve.Retriever method)

C

calculate_hit_and_miss() (in module srtk.retrieve) calculate_subgraph_size() (in module srtk.retrieve) cls_pool() (srtk.scorer.encoder.LitSentenceEncoder static method) Collator (class in srtk.train) compute_embedding_similarity() (srtk.scorer.encoder.LitSentenceEncoder method)

compute_loss() (srtk.scorer.encoder.LitSentenceEncoder method) compute_sentence_similarity() (srtk.scorer.encoder.LitSentenceEncoder method) concate_all() (in module srtk.train) configure_optimizers() (srtk.scorer.encoder.LitSentenceEncoder method) convert_records_relation_id_to_lable() (in module srtk.preprocessing.negative_sampling) create_jsonl_dataset() (in module srtk.preprocessing.negative_sampling)

D

DBpediaLinker (class in srtk.entity_linking.dbpedia) deduce_leaf_relations() (srtk.retrieve.KnowledgeGraphTraverser method) deduce_leaves() (srtk.knowledge_graph.freebase.Freebase method) (srtk.knowledge_graph.graph_base.KnowledgeGraphBase method) (srtk.knowledge_graph.wikidata.Wikidata method) (srtk.retrieve.KnowledgeGraphTraverser method)

deduce_leaves_from_multiple_srcs() (srtk.knowledge_graph.freebase.Freebase method) (srtk.knowledge_graph.wikidata.Wikidata method)

E

ENTITY_PREFIX (srtk.knowledge_graph.wikidata.Wikidata attribute)

expand_and_score_paths() (srtk.retrieve.Retriever method)

F

forward() (srtk.scorer.encoder.LitSentenceEncoder method)

Freebase (class in srtk.knowledge_graph.freebase)

G

generate_paths() (in module srtk.preprocessing.search_path) get_description() (srtk.knowledge_graph.wikidata.Wikidata method) get_entity_label() (srtk.knowledge_graph.graph_base.KnowledgeGraphBase method) get_id_from_uri() (srtk.knowledge_graph.freebase.Freebase static method) get_label() (srtk.knowledge_graph.freebase.Freebase method) (srtk.knowledge_graph.graph_base.KnowledgeGraphBase method) (srtk.knowledge_graph.wikidata.Wikidata method) get_neighbor_relations() (srtk.knowledge_graph.freebase.Freebase method) (srtk.knowledge_graph.graph_base.KnowledgeGraphBase method) (srtk.knowledge_graph.wikidata.Wikidata method)

get_pid_from_uri() (srtk.knowledge_graph.wikidata.Wikidata static method) get_positive_connections_along_paths() (in module srtk.preprocessing.negative_sampling) get_quantifier_filter() (srtk.knowledge_graph.wikidata.Wikidata method) get_relation_label() (srtk.knowledge_graph.freebase.Freebase method) (srtk.knowledge_graph.graph_base.KnowledgeGraphBase method) (srtk.retrieve.KnowledgeGraphTraverser method)

H

has_type_relation() (in module srtk.preprocessing.search_path)

I

is_candidate_space_too_large() (in module srtk.preprocessing.negative_sampling)

is_pid() (srtk.knowledge_graph.wikidata.Wikidata static method) is_qid() (srtk.knowledge_graph.wikidata.Wikidata static method)

K

KnowledgeGraphBase (class in srtk.knowledge_graph.graph_base)

KnowledgeGraphTraverser (class in srtk.retrieve)

L

link() (in module srtk.link)

LinkerBase (class in srtk.entity_linking.linker_base) LitSentenceEncoder (class in srtk.scorer.encoder)

M

main() (in module srtk.preprocessing.load_dataset) (in module srtk.preprocessing.merge_ground) (in module srtk.preprocessing.negative_sampling) (in module srtk.preprocessing.score_path) (in module srtk.preprocessing.search_path) module srtk srtk.entity_linking.dbpedia srtk.entity_linking.linker_base srtk.entity_linking.wikidata srtk.knowledge_graph.freebase srtk.knowledge_graph.graph_base srtk.knowledge_graph.wikidata srtk.link srtk.preprocess srtk.preprocessing.load_dataset srtk.preprocessing.merge_ground srtk.preprocessing.negative_sampling srtk.preprocessing.score_path srtk.preprocessing.search_path srtk.retrieve srtk.scorer.encoder srtk.scorer.scorer srtk.train srtk.visualize

P

Path (class in srtk.retrieve) pool_sentence_embedding() (srtk.scorer.encoder.LitSentenceEncoder method) PREFIXES (srtk.knowledge_graph.freebase.Freebase attribute) (srtk.knowledge_graph.wikidata.Wikidata attribute)

prepare_dataloaders() (in module srtk.train) prev_relations (srtk.retrieve.Path attribute) print_and_save_recall() (in module srtk.retrieve)

Q

queryFreebase() (srtk.knowledge_graph.freebase.Freebase method)

queryWikidata() (srtk.knowledge_graph.wikidata.Wikidata method)

R

retrieve() (in module srtk.retrieve) retrieve_subgraph_triplets() (srtk.retrieve.Retriever method)

Retriever (class in srtk.retrieve) retrive_subgraph() (srtk.retrieve.KnowledgeGraphTraverser method)

S

sample_negative_relations() (in module srtk.preprocessing.negative_sampling) sample_records_from_path() (in module srtk.preprocessing.negative_sampling) save_huggingface_model() (srtk.scorer.encoder.LitSentenceEncoder method) score (srtk.retrieve.Path attribute) score() (srtk.scorer.scorer.Scorer method) score_path() (in module srtk.preprocessing.score_path) Scorer (class in srtk.scorer.scorer) search_one_hop_relations() (srtk.knowledge_graph.freebase.Freebase method) (srtk.knowledge_graph.graph_base.KnowledgeGraphBase method) (srtk.knowledge_graph.wikidata.Wikidata method) search_two_hop_relations() (srtk.knowledge_graph.freebase.Freebase method) (srtk.knowledge_graph.graph_base.KnowledgeGraphBase method) (srtk.knowledge_graph.wikidata.Wikidata method) srtk module srtk.entity_linking.dbpedia module srtk.entity_linking.linker_base module srtk.entity_linking.wikidata module srtk.knowledge_graph.freebase module srtk.knowledge_graph.graph_base module

srtk.knowledge_graph.wikidata module srtk.link module srtk.preprocess module srtk.preprocessing.load_dataset module srtk.preprocessing.merge_ground module srtk.preprocessing.negative_sampling module srtk.preprocessing.score_path module srtk.preprocessing.search_path module srtk.retrieve module srtk.scorer.encoder module srtk.scorer.scorer module srtk.train module srtk.visualize module

T

tokenizer (srtk.train.Collator attribute)

train() (in module srtk.train) training_step() (srtk.scorer.encoder.LitSentenceEncoder method)

V

validation_step() (srtk.scorer.encoder.LitSentenceEncoder method)

visualize() (in module srtk.visualize) visualize_subgraph() (in module srtk.visualize)

W

Wikidata (class in srtk.knowledge_graph.wikidata)

WikidataLinker (class in srtk.entity_linking.wikidata)