Skip to content
Snippets Groups Projects
Commit b2bd3edb authored by alexandre-blanc's avatar alexandre-blanc
Browse files

WIP

parent 06178a29
No related branches found
No related tags found
No related merge requests found
...@@ -70,7 +70,7 @@ if __name__ == '__main__': ...@@ -70,7 +70,7 @@ if __name__ == '__main__':
plt.scatter(embedding[:,0], embedding[:,1], c=[i//M for i in range(len(graph))], cmap='Set1') plt.scatter(embedding[:,0], embedding[:,1], c=[i//M for i in range(len(graph))], cmap='Set1')
plt.arrow(0,0,bias[0], bias[1]) plt.arrow(0,0,bias[0], bias[1])
plt.axis('equal') plt.axis('equal')
writer.add_figure('embedding_end', fig, epoch) writer.add_figure('embedding', fig, epoch)
plt.close() plt.close()
try: try:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment