ixxi-dante / an2vec

Bringing node2vec and word2vec together for cool stuff
GNU General Public License v3.0
22 stars 6 forks source link

Train BlogCatalog without weighing positive/negative errors in adjacency reconstruction loss #38

Closed wehlutyk closed 6 years ago

wehlutyk commented 6 years ago

This tests if the poor adjacency reconstruction we're seeing in #31 comes from the weighing of losses.

wehlutyk commented 6 years ago

The changes made in 3bba9c8fdd5afd66a593a61bf8473b8f794e28fb and 92e08273786a6c3fc1a5e2f8d9358223a74c78fc are for this issue and need to be reverted afterwards. It's now training on grunch, and will be done in about 24h.

wehlutyk commented 6 years ago

(Rerunning as the script picked up the wrong version of the nw2vec package, hardlinked by the environment, instead of the local blogcatalog branch... Back in about 24h)

wehlutyk commented 6 years ago

Done, needs a results notebook.

wehlutyk commented 6 years ago

Results are in f2f38d34917803f88cc1482954c241f2c001493f, see the projects/scale/blogcatalog-dim_ξ=10-no_adj_cross_entropy_weighing-results.ipynb notebook. This change make a big difference!!

Highlights:

Training history

Embedding projections

Adjacency reconstruction

wehlutyk commented 6 years ago

So this means we're not reverting 3bba9c8fdd5afd66a593a61bf8473b8f794e28fb and 92e08273786a6c3fc1a5e2f8d9358223a74c78fc.