OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader Class Reference

Inheritance diagram for OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader:

Collaboration diagram for OpenKE.data.PyTorchTrainDataLoader.PyTorchTrainDataLoader:

Public Member Functions
	__init__ (self, in_path=None, tri_file=None, ent_file=None, rel_file=None, batch_size=None, nbatches=None, threads=8, sampling_mode="normal", bern_flag=False, filter_flag=True, neg_ent=1, neg_rel=0, shuffle=True, drop_last=True)
	set_sampling_mode (self, sampling_mode)
	set_work_threads (self, work_threads)
	set_nbatches (self, nbatches)
	set_batch_size (self, batch_size)
	set_ent_neg_rate (self, rate)
	set_rel_neg_rate (self, rate)
	set_bern_flag (self, bern_flag)
	set_filter_flag (self, filter_flag)
	get_batch_size (self)
	get_ent_tot (self)
	get_rel_tot (self)
	get_triple_tot (self)

Public Attributes
	in_path = in_path
str	tri_file = tri_file
str	ent_file = ent_file
str	rel_file = rel_file
	batch_size = batch_size
	nbatches = nbatches
	num_workers = work_threads

---

The documentation for this class was generated from the following file:

• seed_embeddings/OpenKE/data/PyTorchTrainDataLoader.py