Mirrors/the-algorithm-ml
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm-ml.git synced 2024-12-23 06:41:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update metrics.py

Browse Source 
multiline import
This commit is contained in:
Siddhi Kiran Bajracharyya 2023-04-03 17:59:30 -05:00 committed by GitHub
parent 78c3235eee
commit 415863550d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
core/metrics.py
View File

@ -5,7 +5,11 @@ We assume multi task models will output [task_idx, ...] predictions
"""
from typing import Any, Dict
from tml.core.metric_mixin import MetricMixin, StratifyMixin, TaskMixin
from tml.core.metric_mixin import (
MetricMixin,
StratifyMixin,
TaskMixin
)
import torch
import torchmetrics as tm