/
githubmirror
/
Made-With-ML
Обзор
Документация
Войти
/
githubmirror
/
Made-With-ML
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
tests/code/test_predict.py
11 строк
312 B
GokuMohandas
ML for Developers
26 июл 2023, 14:53
26 июл 2023, 14:53
776a75b
Код
Авторство
О чём код?
from madewithml import predict def test_decode(): decoded = predict.decode(indices=[0, 1, 1], index_to_class={0: "x", 1: "y"}) assert decoded == ["x", "y", "y"] def test_format_prob(): d = predict.format_prob(prob=[0.1, 0.9], index_to_class={0: "x", 1: "y"}) assert d == {"x": 0.1, "y": 0.9}