Skip to content
Snippets Groups Projects
Commit 4c6631de authored by Julius Steiglechner's avatar Julius Steiglechner
Browse files

Include regression metrics.

parent ab294ecc
Branches
No related tags found
No related merge requests found
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Thu Mar 10 12:01:10 2022
@author: jsteiglechner
"""
import torch
def my_mean_squared_error(output: torch.Tensor,
target: torch.Tensor) -> torch.Tensor:
"""Use mean squared error with own function."""
return torch.mean((output - target)**2)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment