_test_migration.py 588 B

12345678910111213141516171819
  1. import sys
  2. sys.path.insert(0, "g:\\TraeWork_pro")
  3. from BiddingKG.dl.rules.loader import RuleLoader
  4. stats = RuleLoader.get_stats()
  5. print("patterns:", stats["pattern_count"])
  6. print("classifiers:", stats["classifier_count"])
  7. print("first 5 pattern_ids:", stats["pattern_ids"][:5])
  8. print("classifier_ids:", stats["classifier_ids"])
  9. # Test a pattern match
  10. m = RuleLoader.match("failure_keywords", "本次招标失败")
  11. print("failure match:", m is not None)
  12. # Test classify
  13. label = RuleLoader.classify("money_source", "财政拨款")
  14. print("money_source classify:", label)
  15. print("ALL OK")