Skip to content

Adding Yaml, Remove args#3

Open
ComTDK wants to merge 10 commits into
kha/topoGED-end-to-endfrom
kha/yaml
Open

Adding Yaml, Remove args#3
ComTDK wants to merge 10 commits into
kha/topoGED-end-to-endfrom
kha/yaml

Conversation

@ComTDK

@ComTDK ComTDK commented Jul 31, 2025

Copy link
Copy Markdown
Collaborator

This PR is to:

  • Add yaml file
  • Remove args
  • Using yaml to assign model config
  • Add Validation step to training loop:
    • Run Validation on the next 10% of the total snapshots
    • Store the best model --> There is self.best_validation_model_auc to compare
  • Resolve some comments:
    1. Return both encoder and decoder after self.train_models()
    2. Sample old nodes in the training loop
  • Add gpu memory alloc

args to yaml:

  • args.embedding --> yaml["encoder_model"]["addOnFeature"]
  • args.embeddingType --> yaml["encoder_model"]["nodeEmbeddingType"]
  • args.mlpEncoding --> yaml["decoder_model"]["encode_links"]

@ComTDK ComTDK requested a review from kia73sha July 31, 2025 18:43
@ComTDK ComTDK self-assigned this Jul 31, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant