rushter--mlalgorithms
Artem Golubin
12 行
285 B
Python
12 行
285 B
Python
from ..metrics import mse, logloss, mae, hinge, binary_crossentropy
|
|
|
|
categorical_crossentropy = logloss
|
|
|
|
|
|
def get_loss(name):
|
|
"""Returns loss function by the name."""
|
|
try:
|
|
return globals()[name]
|
|
except KeyError:
|
|
raise ValueError("Invalid metric function.")
|