cs224u 树结构神经网络PyTorch 实现(6)

cs224u 树结构神经网络PyTorch 实现(6)

__author__ = "Christopher Potts"
__version__ = "CS224u, Stanford, Spring 2020"

构建TorchTreeNN的实例

torch_tree_nn_glove = TorchTreeNN(
    sst_glove_vocab,
    embedding=glove_embedding,
    embed_dim=50,
    max_iter=10,
    eta=0.05

猜你喜欢

转载自blog.csdn.net/duan_zhihua/article/details/106973901