Skip to content

Commit 9c65861

Browse files
committed
split metrics.py into multiple modules.
1 parent 83f194f commit 9c65861

File tree

4 files changed

+749
-4
lines changed

4 files changed

+749
-4
lines changed

src/metrax/__init__.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,15 +12,19 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
from metrax.metrics import (
15+
from metrax.classification_metrics import (
1616
AUCPR,
1717
AUCROC,
18-
MSE,
19-
Perplexity,
2018
Precision,
19+
Recall,
20+
)
21+
from metrax.nlp_metrics import (
22+
Perplexity,
23+
)
24+
from metrax.regression_metrics import (
25+
MSE,
2126
RMSE,
2227
RSQUARED,
23-
Recall,
2428
)
2529

2630
__all__ = [

0 commit comments

Comments
 (0)