ML for Developers

tests/model/utils.py

@@ -0,0 +1,12 @@
+import numpy as np
+import pandas as pd
+
+from madewithml import predict
+
+
+def get_label(text, predictor):
+    df = pd.DataFrame({"title": [text], "description": "", "tag": "other"})
+    z = predictor.predict(data=df)["predictions"]
+    preprocessor = predictor.get_preprocessor()
+    label = predict.decode(np.stack(z).argmax(1), preprocessor.index_to_class)[0]
+    return label