Index

B

backend (RDF2VecConfig attribute), [1], [2]

batch_size (RDF2VecConfig attribute), [1], [2] build_tracker() (in module rdf2vecgpu.logger)

C

CBOW (class in rdf2vecgpu.embedders.word2vec) CBOWDataModule (class in rdf2vecgpu.embedders.word2vec_loader) close() (MlflowTracker method) (WandbTracker method)

construct() (RDF2VecConfig class method), [1], [2] copy() (RDF2VecConfig method), [1], [2] cpu_count (RDF2VecConfig attribute), [1], [2] cudf_to_torch_tensor() (in module rdf2vecgpu.helper.functions)

D

dict() (RDF2VecConfig method), [1], [2]

E

embedding_model (RDF2VecConfig attribute), [1], [2] enabled() (MlflowTracker method) (WandbTracker method)

epochs (RDF2VecConfig attribute), [1], [2]

F

forward() (CBOW method) (OrderAwareCBOW method) (OrderAwareSkipgram method) (SkipGram method)

from_orm() (RDF2VecConfig class method), [1], [2]

G

generate_artifact (RDF2VecConfig attribute), [1], [2] get_context_embeddings() (OrderAwareCBOW method) (OrderAwareSkipgram method)

get_embeddings() (OrderAwareCBOW method) (OrderAwareSkipgram method)

J

json() (RDF2VecConfig method), [1], [2]

K

KGFileReader (class in rdf2vecgpu.reader.kg_file_reader)

L

learning_rate (RDF2VecConfig attribute), [1], [2] literal_bin_strategy (RDF2VecConfig attribute), [1], [2] literal_n_bins (RDF2VecConfig attribute), [1], [2] literal_predicates (RDF2VecConfig attribute), [1], [2] literal_strategy (RDF2VecConfig attribute), [1], [2] log_artifact() (MlflowTracker method) (WandbTracker method) log_data() (MlflowTracker method) (WandbTracker method) log_dict() (WandbTracker method)

log_figure() (MlflowTracker method) (WandbTracker method) log_metrics() (MlflowTracker method) (WandbTracker method) log_model_pytorch() (MlflowTracker method) (WandbTracker method) log_params() (MlflowTracker method) (WandbTracker method) log_pytorch() (MlflowTracker method) (WandbTracker method) log_text() (WandbTracker method)

M

min_count (RDF2VecConfig attribute), [1], [2] MlflowTracker (class in rdf2vecgpu.logger.mlflow_logger) model_computed_fields (RDF2VecConfig attribute), [1], [2] model_config (RDF2VecConfig attribute), [1], [2] model_construct() (RDF2VecConfig class method), [1], [2] model_copy() (RDF2VecConfig method), [1], [2] model_dump() (RDF2VecConfig method), [1], [2] model_dump_json() (RDF2VecConfig method), [1], [2] model_extra (RDF2VecConfig property), [1], [2] model_fields (RDF2VecConfig attribute), [1], [2] model_fields_set (RDF2VecConfig property), [1], [2] model_json_schema() (RDF2VecConfig class method), [1], [2] model_parametrized_name() (RDF2VecConfig class method), [1], [2] model_post_init() (RDF2VecConfig method), [1], [2] model_rebuild() (RDF2VecConfig class method), [1], [2] model_validate() (RDF2VecConfig class method), [1], [2] model_validate_json() (RDF2VecConfig class method), [1], [2]

model_validate_strings() (RDF2VecConfig class method), [1], [2] module rdf2vecgpu rdf2vecgpu.config rdf2vecgpu.corpus rdf2vecgpu.embedders rdf2vecgpu.embedders.word2vec rdf2vecgpu.embedders.word2vec_loader rdf2vecgpu.helper rdf2vecgpu.helper.functions rdf2vecgpu.logger rdf2vecgpu.logger.mlflow_logger rdf2vecgpu.logger.wandb_logger rdf2vecgpu.reader rdf2vecgpu.reader.kg_file_reader src multi_gpu (RDF2VecConfig attribute), [1], [2]

N

negative_samples (RDF2VecConfig attribute), [1], [2]

num_nodes (RDF2VecConfig attribute), [1], [2]

O

OrderAwareCBOW (class in rdf2vecgpu.embedders.word2vec) OrderAwareCBOWDataModule (class in rdf2vecgpu.embedders.word2vec_loader)

OrderAwareSkipgram (class in rdf2vecgpu.embedders.word2vec) OrderAwareSkipGramDataModule (class in rdf2vecgpu.embedders.word2vec_loader)

P

ParquetSkipGramDataModule (class in rdf2vecgpu.embedders.word2vec_loader) parse_file() (RDF2VecConfig class method), [1], [2]

parse_obj() (RDF2VecConfig class method), [1], [2] parse_raw() (RDF2VecConfig class method), [1], [2] predict_center() (OrderAwareCBOW method)

R

random_state (RDF2VecConfig attribute), [1], [2] RDF2VecConfig (class in rdf2vecgpu) (class in rdf2vecgpu.config), [1] rdf2vecgpu module rdf2vecgpu.config module rdf2vecgpu.corpus module rdf2vecgpu.embedders module rdf2vecgpu.embedders.word2vec module rdf2vecgpu.embedders.word2vec_loader module

rdf2vecgpu.helper module rdf2vecgpu.helper.functions module rdf2vecgpu.logger module rdf2vecgpu.logger.mlflow_logger module rdf2vecgpu.logger.wandb_logger module rdf2vecgpu.reader module rdf2vecgpu.reader.kg_file_reader module read() (KGFileReader method) reproducible (RDF2VecConfig attribute), [1], [2]

S

schema() (RDF2VecConfig class method), [1], [2] schema_json() (RDF2VecConfig class method), [1], [2] set_tags() (MlflowTracker method) (WandbTracker method) setup() (CBOWDataModule method) (ParquetSkipGramDataModule method) (SkipGramDataModule method)

SkipGram (class in rdf2vecgpu.embedders.word2vec) SkipGramDataModule (class in rdf2vecgpu.embedders.word2vec_loader) src module stage() (MlflowTracker method) (WandbTracker method) start_pipeline() (MlflowTracker method) (WandbTracker method)

T

torch_to_cudf() (in module rdf2vecgpu.helper.functions) tracker (RDF2VecConfig attribute), [1], [2] tracker_kwargs (RDF2VecConfig attribute), [1], [2] tracker_run_name (RDF2VecConfig attribute), [1], [2]

training_step() (CBOW method) (OrderAwareCBOW method) (OrderAwareSkipgram method) (SkipGram method) tune_batch_size (RDF2VecConfig attribute), [1], [2]

U

update_forward_refs() (RDF2VecConfig class method), [1], [2]

V

validate() (RDF2VecConfig class method), [1], [2]

vector_size (RDF2VecConfig attribute), [1], [2]

W

walk_depth (RDF2VecConfig attribute), [1], [2] walk_number (RDF2VecConfig attribute), [1], [2] walk_strategy (RDF2VecConfig attribute), [1], [2]

walk_weighted (RDF2VecConfig attribute), [1], [2] WandbTracker (class in rdf2vecgpu.logger.wandb_logger) window_size (RDF2VecConfig attribute), [1], [2]