quaterion package

Subpackages

• quaterion.dataset package
  • Submodules
    • quaterion.dataset.indexing_dataset module
      • IndexingDataset
      • IndexingIterableDataset
    • quaterion.dataset.label_cache_dataset module
      • LabelCacheDataset
      • LabelCacheDatasetMixin
      • LabelCacheIterableDataset
      • LabelCacheMode
    • quaterion.dataset.similarity_data_loader module
      • GroupSimilarityDataLoader
      • PairsSimilarityDataLoader
      • SimilarityDataLoader
    • quaterion.dataset.similarity_dataset module
      • SimilarityGroupDataset
    • quaterion.dataset.similarity_samples module
      • SimilarityGroupSample
      • SimilarityPairSample
    • quaterion.dataset.train_collator module
      • TrainCollator
  • Module contents
• quaterion.distances package
  • Submodules
    • quaterion.distances.base_distance module
      • BaseDistance
    • quaterion.distances.cosine module
      • Cosine
    • quaterion.distances.dot_product module
      • DotProduct
    • quaterion.distances.euclidean module
      • Euclidean
    • quaterion.distances.manhattan module
      • Manhattan
  • Module contents
    • Distance
      • Distance.get_by_name()
      • Distance.COSINE
      • Distance.DOT_PRODUCT
      • Distance.EUCLIDEAN
      • Distance.MANHATTAN
• quaterion.eval package
  • Subpackages
    • quaterion.eval.accumulators package
      • Submodules
      • Module contents
    • quaterion.eval.group package
      • Submodules
      • Module contents
    • quaterion.eval.pair package
      • Submodules
      • Module contents
    • quaterion.eval.samplers package
      • Submodules
      • Module contents
  • Submodules
    • quaterion.eval.attached_metric module
      • AttachedMetric
    • quaterion.eval.base_metric module
      • BaseMetric
    • quaterion.eval.evaluator module
      • Evaluator
  • Module contents
• quaterion.loss package
  • Subpackages
    • quaterion.loss.extras package
      • Submodules
      • Module contents
  • Submodules
    • quaterion.loss.arcface_loss module
      • ArcFaceLoss
    • quaterion.loss.center_loss module
      • CenterLoss
    • quaterion.loss.circle_loss module
      • CircleLoss
    • quaterion.loss.contrastive_loss module
      • ContrastiveLoss
    • quaterion.loss.cos_face_loss module
      • CosFaceLoss
    • quaterion.loss.fast_ap_loss module
      • FastAPLoss
    • quaterion.loss.group_loss module
      • GroupLoss
    • quaterion.loss.multiple_negatives_ranking_loss module
      • MultipleNegativesRankingLoss
    • quaterion.loss.online_contrastive_loss module
      • OnlineContrastiveLoss
    • quaterion.loss.pairwise_loss module
      • PairwiseLoss
    • quaterion.loss.similarity_loss module
      • SimilarityLoss
    • quaterion.loss.softmax_loss module
      • SoftmaxLoss
    • quaterion.loss.triplet_loss module
      • TripletLoss
  • Module contents
• quaterion.train package
  • Subpackages
    • quaterion.train.cache package
      • Submodules
      • Module contents
    • quaterion.train.callbacks package
      • Submodules
      • Module contents
    • quaterion.train.xbm package
      • Submodules
      • Module contents
  • Submodules
    • quaterion.train.cache_mixin module
      • CacheMixin
    • quaterion.train.trainable_model module
      • TrainableModel
  • Module contents
• quaterion.utils package
  • Submodules
    • quaterion.utils.enums module
      • TrainStage
    • quaterion.utils.progress_bar module
      • FixedLengthProcessionSpeed
      • QuaterionProgressBar
    • quaterion.utils.utils module
      • get_anchor_negative_mask()
      • get_anchor_positive_mask()
      • get_masked_maximum()
      • get_masked_minimum()
      • get_triplet_mask()
      • info_value_of_dtype()
      • iter_by_batch()
      • l2_norm()
      • max_value_of_dtype()
      • min_value_of_dtype()
  • Module contents

Submodules

• quaterion.main module
  • Quaterion
    • Quaterion.evaluate()
    • Quaterion.fit()
    • Quaterion.trainer_defaults()

Module contents