【Python】查看划分数据集具体大小

1、动机

 

一般一些经典的数据集有人处理好，则我们直接拿来用，但是有时候不知道具体数据集划分的大小，这个时候如果想查看的话，便需要统计一下了

2、查看

一般mask就是bool变量，我的代码中是一维张量，如果想具体节点大小的话，则可以通过下面方式

    train_idx = g.nodes[predict_ntype].data['train_mask']
    val_idx = g.nodes[predict_ntype].data['val_mask']
    test_idx = g.nodes[predict_ntype].data['test_mask']

    
    print(train_idx.shape)
    print('True个数：', sum(train_idx!=0))
    print('True个数：', sum(val_idx!=0))
    print('True个数：', sum(test_idx!=0))

结果 

3、参考

Python计算一个数组中True或False的个数

Python数据处理入门教程！