def __getitem__()

in domainbed_measures/measures/registry.py [0:0]

• 43 lines of code
• 27 McCabe index (conditional complexity)

    def __getitem__(self, measure="entropy"):
        if measure not in self._VALID_MEASURES:
            raise NotImplementedError

        if measure == "entropy" or measure == 'entropy_held_out':
            return EntropyOutput
        elif measure == "path_norm":
            return PathNorm
        elif measure == "sharp_mag":
            return SharpnessMagnitude
        elif (measure == "v_minimality" or measure == "v_plus_minimality"
              or measure == "cond_v_minimality"
              or measure == "cond_v_plus_minimality"):
            return FunctionVMinimality
        elif measure == "jacobian_norm":
            return JacobianNorm
        elif measure == "jacobian_norm_relative":
            return JacobianNormRelative
        elif measure == "jacobian_norm_relative_log_diff":
            return JacobianNormRelativeLogDiff
        elif measure == "jacobian_norm_relative_diff":
            return JacobianNormRelativeDiff
        elif measure == "validation_accuracy":
            return ValidationAccuracy
        elif measure == "mixup" or measure == "mixup_alpha_0.3":
            return Mixup
        elif measure == "mixup_relative" or measure == "mixup_alpha_0.3_relative":
            return MixupRelative
        elif measure == "mixup_relative_diff":
            return MixupRelativeDiff
        elif measure == "mixup_relative_log_diff":
            return MixupRelativeLogDiff
        elif "fisher_eigval_sum_diff" in measure:
            return FisherEigValuesSumDiff
        elif "fisher_eigval_sum" in measure:
            return FisherEigValues
        elif "fisher_eigvec_align" in measure:
            return FisherEigVecAlign
        elif "c2st" in measure:
            return ClassifierTwoSampleTest
        elif "mmd" in measure:
            return MMD
        elif 'hdh' in measure:
            return HDelHDivergence