deploy: 0751985

Author: melvinbarbaux

data/artifacts/00022a5cb284c7d3e995/config.yaml

@@ -0,0 +1,99 @@
+run:
+  name: best_text_transductive_gat_dbpedia_14_R6-seed3
+  seed: 3
+  output_dir: /lustre/fsn1/projects/rech/dqp/uqj57xk/projet_modssc/outputs/R6/transductive/gat/text/dbpedia_14/best_text_transductive_gat_dbpedia_14_R6-sweep-20260407T073920Z
+  log_level: detailed
+  fail_fast: true
+limits:
+  profile: auto
+dataset:
+  id: dbpedia_14
+  download: true
+  cache_dir: /lustre/fsn1/projects/rech/dqp/uqj57xk/projet_modssc/modssc_cache/datasets
+  options:
+    text_column: content
+    label_column: label
+    prefer_test_split: true
+sampling:
+  seed: 3
+  plan:
+    split:
+      kind: holdout
+      test_fraction: 0.2
+      val_fraction: 0.1
+      stratify: true
+      shuffle: true
+    labeling:
+      mode: per_class
+      value: 50
+      strategy: balanced
+      min_per_class: 1
+      per_class: true
+      fixed_indices: null
+    imbalance:
+      kind: none
+    policy:
+      respect_official_test: true
+      use_official_graph_masks: true
+      allow_override_official: false
+preprocess:
+  seed: 3
+  fit_on: train_labeled
+  cache: true
+  cache_dir: /lustre/fsn1/projects/rech/dqp/uqj57xk/projet_modssc/modssc_cache/preprocess
+  plan:
+    output_key: features.X
+    steps:
+    - id: labels.encode
+    - id: text.ensure_strings
+    - id: text.sentence_transformer
+      params:
+        batch_size: 32
+        model_id: st:all-MiniLM-L6-v2
+    - id: core.pca
+      params:
+        n_components: 128
+    - id: core.to_numpy
+method:
+  kind: transductive
+  id: gat
+  device:
+    device: auto
+    dtype: float32
+    resolved_device: cuda
+  params:
+    head_dim: 8
+    heads: 8
+    dropout: 0.6
+    negative_slope: 0.2
+    lr: 0.005
+    weight_decay: 0.0005
+    max_epochs: 200
+    patience: 100
+    add_self_loops: true
+evaluation:
+  split_for_model_selection: val
+  report_splits:
+  - val
+  - test
+  metrics:
+  - accuracy
+  - macro_f1
+graph:
+  enabled: true
+  seed: 3
+  cache: true
+  cache_dir: /lustre/fsn1/projects/rech/dqp/uqj57xk/projet_modssc/modssc_cache/graph
+  spec:
+    scheme: knn
+    metric: cosine
+    k: 15
+    symmetrize: mutual
+    weights:
+      kind: heat
+      sigma: 1.0
+    normalize: rw
+    self_loops: true
+    backend: numpy
+    chunk_size: 1024
+    feature_field: features.X