| """Predict banking intents with the adjacent local model.skops artifact.""" | |
| import argparse | |
| import json | |
| from pathlib import Path | |
| import skops.io as sio | |
| def main(): | |
| parser = argparse.ArgumentParser(description=__doc__) | |
| parser.add_argument("text", nargs="+") | |
| parser.add_argument("--model", type=Path, default=Path(__file__).with_name("model.skops")) | |
| args = parser.parse_args() | |
| unexpected = sio.get_untrusted_types(file=args.model) | |
| if unexpected: | |
| raise RuntimeError(f"Unexpected model types: {unexpected}") | |
| model = sio.load(args.model, trusted=[]) | |
| probabilities = model.predict_proba(args.text) | |
| for text, probability in zip(args.text, probabilities): | |
| order = sorted(range(len(model.classes_)), key=lambda i: (-probability[i], model.classes_[i]))[:3] | |
| print(json.dumps({"text": text, "top3": [{"label": str(model.classes_[i]), "score": float(probability[i])} for i in order]})) | |
| if __name__ == "__main__": | |
| main() | |