Metrics

Metrics

class torchmimic.metrics.AUCROC(average=None)

AUCROC scoring class

class torchmimic.metrics.AverageMeter

Class used to collect values and return a running average

reset()

Resets private members

update(val, _n=1)

Updates class members

Parameters
  • val (float) – value used to update running average

  • _n (int) – sample size used to calculate value

class torchmimic.metrics.MetricMeter(score_fn)

Class used to collect values and evaluate them using a scoring function

reset()

Resets private members

score()

Scores true and predicted values :returns: the output of the score function given the predicted and true labels :rtype: int

update(true, pred)

Updates list of true and predicted values

Parameters
  • true (np.array) – true labels

  • pred (np.array) – predicted labels

torchmimic.metrics.accuracy(true, pred)

Returns the accuracy for the provided true and predicted values

Parameters
  • true (np.array) – true values

  • pred (np.array) – predicted values

Returns

accuracy score

Return type

int

torchmimic.metrics.aucpr(true, pred)

Returns the AUC-PR for the provided true and predicted values

Parameters
  • true (np.array) – true values

  • pred (np.array) – predicted values

Returns

AUC-PR score

Return type

int

torchmimic.metrics.balanced_accuracy(true, pred)

Returns the Balanced Accuracy for the provided true and predicted values

Parameters
  • true (np.array) – true values

  • pred (np.array) – predicted values

Returns

Balanced Accuracy score

Return type

int

torchmimic.metrics.f1(true, pred)

Returns the F1-score for the provided true and predicted values

Parameters
  • true (np.array) – true values

  • pred (np.array) – predicted values

Returns

F1-score

Return type

int

torchmimic.metrics.kappa(true, pred)

Returns the Cohen’s Kappa for the provided true and predicted values

Parameters
  • true (np.array) – true values

  • pred (np.array) – predicted values

Returns

Cohen’s Kappa score

Return type

int

torchmimic.metrics.mae(true, pred)

Returns the Mean Absolute Error/Deviation for the provided true and predicted values

Parameters
  • true (np.array) – true values

  • pred (np.array) – predicted values

Returns

MAE/MAD score

Return type

int