config.py 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. # 表头识别模型
  2. ai2config = {
  3. "table_field_config": {
  4. "corpus_path": "./docs/table_head_doc/field_label.csv",
  5. "vocab_file": './docs/table_head_doc/table_head_vocab',
  6. "vocab_label": './docs/table_head_doc/table_head_le',
  7. "epochs": 4000,
  8. "lr": 1e-2,
  9. "momentum": .8,
  10. "model_path": './docs/table_head_doc/table_head_clf_model'
  11. }}
  12. amount_config = {
  13. "table_field_config": {
  14. "path": "./docs/table_head_doc/amount_label.csv",
  15. }}
  16. budget_config = {
  17. "table_field_config": {
  18. "path": "./docs/table_head_doc/budget_label.csv",
  19. }}
  20. general_config = {
  21. "table_field_config": {
  22. "path": "./docs/table_head_doc/general_label.csv",
  23. }}
  24. abnormal_config = {
  25. "table_field_config": {
  26. "path1": "./docs/table_head_doc/abnormal_buyer_begin.csv",
  27. "path2": "./docs/table_head_doc/abnormal_buyer_contain.csv",
  28. "path3": "./docs/table_head_doc/abnormal_buyer_end.csv",
  29. "path4": "./docs/table_head_doc/projectcode.csv",
  30. "path5": "./docs/table_head_doc/title_abnormal.csv",
  31. "path6": "./docs/table_head_doc/title_abnormal_contain.csv",
  32. "path7": "./docs/table_head_doc/trusted_site.csv"
  33. }}
  34. # 调试
  35. DEBUG = True
  36. # 字段识别模型
  37. FieldServer = {
  38. "a2s_ip": "172.17.0.11:9090",
  39. "topic": "goods_field",
  40. "timeout": 60,
  41. "retry_times": 5,
  42. }
  43. # 组织机构识别模型
  44. RecognitionServer = {
  45. "a2s_ip": "172.17.0.11:9090",
  46. "topic": "recognition_org",
  47. "timeout": 60,
  48. "retry_times": 5,
  49. }
  50. # 产品识别模型
  51. ProductServer = {
  52. "a2s_ip": "172.17.0.11:9090",
  53. "topic": "recognition_goods",
  54. "timeout": 60,
  55. "retry_times": 5,
  56. }
  57. # update_urls_mongodb
  58. ReluMongodb = {
  59. "ip_port": "192.168.3.71:29099",
  60. "user": "",
  61. "password": "",
  62. "db": "re4art",
  63. "col": "quality",
  64. }