ML for Developers

tests/code/test_predict.py

@@ -0,0 +1,11 @@
+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}