python-chebifier/chebifier/model_registry.yml at e10e12981a9ed0ea572a5ff5a9b1449c0e3b5044 · ChEB-AI/python-chebifier · 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
electra_chebi50-3star_v244:
  type: electra
  hugging_face:
    repo_id: chebai/electra_chebi50-3star_v244
    files:
      ckpt_path: electra_chebi50-3star_v244_x2mngani_epoch=180.ckpt
      classwise_weights_path: electra_chebi50-3star_v244_x2mngani_epoch=180_trust_3star.json
gat_chebi50_v244:
  type: gat
  hugging_face:
    repo_id: chebai/gat_chebi50_v244
    files:
      ckpt_path: gat_chebi50_v244_0nfi19qt_epoch=198.ckpt
      classwise_weights_path: gat_chebi50_v244_0nfi19qt_epoch=198_trust_3star.json
gat-aug_chebi50_v244:
  type: gat
  hugging_face:
    repo_id: chebai/gat-aug_chebi50_v244
    files:
      ckpt_path: gat-aug_chebi50_v244_8fky8tru_epoch=192.ckpt
      classwise_weights_path: gat-aug_chebi50_v244_8fky8tru_epoch=192_trust_3star.json
resgated-aug_chebi50-3star_v244:
  type: resgated
  hugging_face:
    repo_id: chebai/resgated-aug_chebi50-3star_v244
    files:
      ckpt_path: resgated-aug_chebi50-3star_v244_w0rhmajx_epoch=190.ckpt
      classwise_weights_path: resgated-aug_chebi50-3star_v244_w0rhmajx_epoch=190_trust_3star.json
electra_chebi50_v241:
    type: electra
    hugging_face:
      repo_id: chebai/electra_chebi50_v241
      files:
        ckpt_path: 14ko0zcf_epoch=193.ckpt
        classwise_weights_path: metrics_electra_14ko0zcf_80-10-10_short.json
resgated_chebi50_v241:
    type: resgated
    hugging_face:
      repo_id: chebai/resgated_gcn_chebi50_v241
      files:
        ckpt_path: 0ps1g189_epoch=122.ckpt
        classwise_weights_path: metrics_0ps1g189_80-10-10_short.json
c3p_with_weights:
  type: c3p
  hugging_face:
    repo_id: chebai/chebifier
    repo_type: dataset
    files:
      classwise_weights_path: c3p_trust.json