Index

Symbols

--version stonkgs command line option

B

BertForPreTrainingOutputWithPooling (class in stonkgs.models.stonkgs_model)

C

configure_optimizers() (KGEClassificationModel method)

F

forward() (KGEClassificationModel method) (STonKGsELMPredictionHead method) (STonKGsForPreTraining method) (STonKGsForSequenceClassification method)

from_default_pretrained() (STonKGsForPreTraining class method)

G

get_embeddings() (Node2VecINDRAEntityDataset method) (TransEINDRAEntityDataset method)

get_train_test_splits() (in module stonkgs.models.kg_baseline_model) (in module stonkgs.models.nlp_baseline_model) (in module stonkgs.models.stonkgs_finetuning)

I

INDRADataset (class in stonkgs.models.stonkgs_finetuning)

INDRAEvidenceDataset (class in stonkgs.models.nlp_baseline_model)

K

KGEClassificationModel (class in stonkgs.models.kg_baseline_model)

M

module stonkgs.models.kg_baseline_model stonkgs.models.nlp_baseline_model stonkgs.models.stonkgs_finetuning stonkgs.models.stonkgs_model stonkgs.models.stonkgs_pretraining

N

Node2VecINDRAEntityDataset (class in stonkgs.models.kg_baseline_model)

P

prepare_df() (in module stonkgs.models.kg_baseline_model)

preprocess_fine_tuning_data() (in module stonkgs.models.stonkgs_finetuning)

R

run_kg_baseline_classification_cv() (in module stonkgs.models.kg_baseline_model)

run_nlp_baseline_classification_cv() (in module stonkgs.models.nlp_baseline_model) run_sequence_classification_cv() (in module stonkgs.models.stonkgs_finetuning)

S

stonkgs command line option --version stonkgs.models.kg_baseline_model module stonkgs.models.nlp_baseline_model module stonkgs.models.stonkgs_finetuning module

stonkgs.models.stonkgs_model module stonkgs.models.stonkgs_pretraining module STonKGsELMPredictionHead (class in stonkgs.models.stonkgs_model) STonKGsForPreTraining (class in stonkgs.models.stonkgs_model) STonKGsForSequenceClassification (class in stonkgs.models.stonkgs_finetuning)

T

test_epoch_end() (KGEClassificationModel method) test_step() (KGEClassificationModel method)

training_step() (KGEClassificationModel method) TransEINDRAEntityDataset (class in stonkgs.models.kg_baseline_model)