serf/config.yml at main · Graphlet-AI/serf · GitHub

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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
logs:
  file:
    path: logs

models:
  embedding: "intfloat/multilingual-e5-base"
  llm: "gemini/gemini-2.0-flash"
  analyze_llm: "${models.llm}"
  temperature: 0.0

er:
  blocking:
    method: semantic
    target_block_size: 30
    max_block_size: 100
    min_block_size: 2
    auto_scale_by_iteration: true

  matching:
    batch_size: 10
    max_concurrent: 20
    temperature: 0.0
    max_retries: 3
    retry_delay_ms: 300

  eval:
    coverage_threshold: 0.9999
    error_threshold: 0.0001
    overlap_threshold: 0.01

  paths:
    blocks: "data/iteration_{iteration}/blocks"
    matches: "data/iteration_{iteration}/matches"
    resolved: "data/iteration_{iteration}/resolved"
    edges: "data/iteration_{iteration}/edges"
    metrics: "data/iteration_{iteration}/metrics"

mlflow:
  tracking_uri: "http://127.0.0.1:5001"
  experiment_name: "SERF-Entity-Resolution"
  backend_store_uri: "sqlite:///data/mlflow.db"
  host: "127.0.0.1"
  port: 5001
  autolog:
    log_traces: true
    log_traces_from_compile: false
    log_traces_from_eval: true

benchmarks:
  output_dir: "data/benchmarks"
  datasets:
    walmart-amazon:
      url: "https://pages.cs.wisc.edu/~anhai/data1/deepmatcher_data/Structured/Walmart-Amazon/walmart_amazon_exp_data.zip"
      domain: products
      difficulty: hard
    abt-buy:
      url: "https://pages.cs.wisc.edu/~anhai/data1/deepmatcher_data/Textual/Abt-Buy/exp_data"
      domain: products
      difficulty: hard
    amazon-google:
      url: "https://pages.cs.wisc.edu/~anhai/data1/deepmatcher_data/Structured/Amazon-Google/amazon_google_exp_data.zip"
      domain: products
      difficulty: hard
    dblp-acm:
      url: "https://pages.cs.wisc.edu/~anhai/data1/deepmatcher_data/Structured/DBLP-ACM/dblp_acm_exp_data.zip"
      domain: bibliographic
      difficulty: easy
    dblp-scholar:
      url: "https://pages.cs.wisc.edu/~anhai/data1/deepmatcher_data/Structured/DBLP-GoogleScholar/dblp_scholar_exp_data.zip"
      domain: bibliographic
      difficulty: medium