pytorch_DGCNN icon indicating copy to clipboard operation
pytorch_DGCNN copied to clipboard

About feature

Open huizhang1032 opened this issue 3 years ago • 5 comments

How to output the feature of the last layer of the neural network of each graph in the list of graphs to be tested ?

huizhang1032 avatar Jul 18 '21 02:07 huizhang1032

You can add a "-extract_features" to your command to save the test graphs' embeddings to "extracted_features_test.txt". See this line.

muhanzhang avatar Jul 18 '21 08:07 muhanzhang

OK.Thank you!

huizhang1032 avatar Jul 18 '21 12:07 huizhang1032

If I input 100 graph objects to the model, the output feature is 10032. Does 132 here represent the embed of a graph object?

huizhang1032 avatar Jul 23 '21 08:07 huizhang1032

Hi, if you input 100 graphs, and set the final graph embedding size to 32, then you will get 100 * (1 + 32) dimensional matrix in your output txt. The 1 additional dimension corresponds to the predicted labels of all input graphs.

muhanzhang avatar Jul 25 '21 03:07 muhanzhang

I got it.Thank you very much!

huizhang1032 avatar Jul 28 '21 01:07 huizhang1032