logitorch.data_collators.ruletaker_collator

Classes

RuleTakerCollator	A collator class for RuleTaker model.
RuleTakerProofWriterCollator	A collator class for RuleTaker with ProofWriter model.

Module Contents

class logitorch.data_collators.ruletaker_collator.RuleTakerCollator

A collator class for RuleTaker model.

This collator is used to preprocess and collate data for RuleTaker model training or inference.

tokenizer

class logitorch.data_collators.ruletaker_collator.RuleTakerProofWriterCollator

A collator class for RuleTaker with ProofWriter model.

This collator is used to preprocess and collate data for RuleTaker with ProofWriter model training or inference.

tokenizer