About node embedding function

[wubo2180]

Hi, could you please provide the function that returns the trained node embeddings so that I can input the embeddings to machine learning classifier such as SVM.

Best wish!

[kayzliu]

Hi! Thanks for your issue. Maybe we can add a parameter (e.g., model, emb = model.fit(G, return_emb=True)) for AE-based detectors in future version. Does it look good to you?

[wubo2180]

Thanks a lot! That is what I expect.

[kayzliu]

The input data embedding is now available for deep learning based methods via detector.emb. During inference, you can also retrieve the embedding by red, emb = detector.predict(test_data, return_emb=True)