search

graphsinspace / graspe

Graphs in Space: Graph Embeddings for Machine Learning on Complex Data
GNU General Public License v3.0
7 stars 0 forks source link

LID-aware GCN Embedding #19

Closed nmilosev closed 3 years ago

nmilosev commented 3 years ago

Prva verzija GCN embeddinga koja pokušava da minimizuje LID

Molim @milsav da pogleda kad stigne.

  1. u fajlu lid_eval je dodata EmbLIDMLEEstimatorTorch klasa koja radi isto kao EmbLIDMLEEstimator klasa ali radi sa torch funkcijama i pamti gradijente
  2. dodat je test test_gcn_lid koji testira da li novi estimator radi korektno (poredim sa "normalnom" verzijom)
  3. u fajlu embedding_gcn.py dodata su dva parametra lid_aware i lid_k. Kod optimizacije je dodat dodatni kriterijum kako bi se pored pravljenja korektnog embedding-a minimizovala LID vrednost (koristim sumu lid vrednosti)
  4. dodat je test test_lid_aware_gcn koji testira novi embedding, poredi LID vrednosti normalnog i novog i recon loss oba embedding-a

Rezultati deluju ok, treba testirati za veći graf:

lid sum (LID-Aware) tensor(613.4803)
recon loss (LID-Aware) 0.3

lid sum (normal) tensor(649.9783)
recon loss (normal) 0.0
milsav commented 3 years ago

@nmilosev -- pogledao, sve deluje ok