Some questions about some parts of the code in file(open_word_prediction.py , train())

[pigcowherd]

hi, haseebs, here is my question:

in line 342: """ iterator = self.dataset.train.iter_entitydata_entitywise num_batches = self.dataset.train.num_entities // batch_size

    for epoch in range(start_epoch, epochs + 1):
        train_batch_losses = []
        decay_lr(current_epoch=epoch, optimizer=self.optimizer)

        for entities, names, descriptions, relation_ids in tqdm(iterator(batch_size,
                                                         yield_heads=True,
                                                         yield_tails=Config.get("UseTailsToOptimize"),
                                                         cluster_relations=Config.get("ClusterRelations")),
                                                total=num_batches,
                                                desc="Train: "):

"""

"iterator= self.dataset.train.iter_entitydata_entitywise", function "iter_entitydata_entitywise" only return entities, names, descriptions, ,there is not relations id.

[pigcowherd]

and I changed it to: iterator = self.dataset.train.iter_entitydata_triplewise