dgs.models.similarity.similarity.SimilarityModule.get_train_data

SimilarityModule.get_train_data(ds: State) any[source]

A custom function to get special data for training purposes. If “train_key” is not given, uses the regular get_data() function of this module.