Source code for geowatch.monkey.monkey_torchmetrics


[docs]
def fix_torchmetrics_compatability():
    import torchmetrics
    from geowatch.monkey._monkey_fbeta import FBetaScore_Patched
    from geowatch.monkey._monkey_fbeta import Accuracy_Patched
    f_beta = torchmetrics.classification.f_beta
    accuracy = torchmetrics.classification.accuracy

    if not hasattr(f_beta, 'FBetaScoreOrig'):
        f_beta.FBetaScoreOrig = f_beta.FBetaScore

    if not hasattr(accuracy, 'FBetaScoreOrig'):
        accuracy.AccuracyOrig = accuracy.Accuracy

    f_beta.FBetaScore = FBetaScore_Patched
    f_beta.F1 = FBetaScore_Patched

    accuracy.Accuracy = Accuracy_Patched