File size: 636 Bytes
71d49e7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | import os
import sys
ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, ROOT)
from skala import SkalaClassifier
def main() -> int:
default = "I have this Python codebase, can you port it to C++"
text = " ".join(sys.argv[1:]) or default
classifier = SkalaClassifier(os.path.join(ROOT, "artifacts"))
classifier.load()
classifier.warmup()
label, scores = classifier.classify(text)
print(f"{text!r} -> {label}")
for name in ("simple", "medium", "complex"):
print(f" {name}: {scores[name]:.3f}")
return 0
if __name__ == "__main__":
sys.exit(main())
|