Skip to content

Instantly share code, notes, and snippets.

@larsbratholm
Created February 7, 2020 16:01
Show Gist options
  • Save larsbratholm/81528a3c64596c0b385d4784490b5680 to your computer and use it in GitHub Desktop.
Save larsbratholm/81528a3c64596c0b385d4784490b5680 to your computer and use it in GitHub Desktop.
if metric == 'rmse':
metric_fun = lambda x: np.std(x, axis=1)
elif metric == 'mae':
metric_fun = lambda x: np.mean(abs(x), axis=1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment