search

Tomorrowdawn / EvolveGCN

Semester project for social computing of USTC.
0 stars 3 forks source link

测试项目 #4

Open Tomorrowdawn opened 2 months ago

Tomorrowdawn commented 2 months ago

model相关

测试model/文件夹下面的类, AttentionPooling, GATPredictor, WeightedGAT. 检查给定预期输入后是否产生预期的输出, 维度是否正确.

方法: def get_embeddings(embeddings, *indices): 检查是否按照indices取出embedding, indices=-1时是否取出的是0 embedding.