unsupervised_train.py 文件源码

python
阅读 20 收藏 0 点赞 0 评论 0

项目:GraphSAGE 作者: williamleif 项目源码 文件源码
def construct_placeholders():
    # Define placeholders
    placeholders = {
        'batch1' : tf.placeholder(tf.int32, shape=(None), name='batch1'),
        'batch2' : tf.placeholder(tf.int32, shape=(None), name='batch2'),
        # negative samples for all nodes in the batch
        'neg_samples': tf.placeholder(tf.int32, shape=(None,),
            name='neg_sample_size'),
        'dropout': tf.placeholder_with_default(0., shape=(), name='dropout'),
        'batch_size' : tf.placeholder(tf.int32, name='batch_size'),
    }
    return placeholders
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号