mirror of
https://github.com/twitter/the-algorithm-ml.git
synced 2024-05-19 22:57:16 +01:00
8 lines
146 B
Python
8 lines
146 B
Python
"""Loss type enums."""
|
|
from enum import Enum
|
|
|
|
|
|
class LossType(str, Enum):
|
|
CROSS_ENTROPY = "cross_entropy"
|
|
BCE_WITH_LOGITS = "bce_with_logits"
|