mayubo2333/fewshot_ed

ACL'2023: Few-shot Event Detection: An Empirical Study and a Unified View

fewshot_ed's Issues

使用MAVEN数据集运行时出现AssertionError

 tok_embeddings.shape[0]: 97315
event_feature_num 97309
Traceback (most recent call last):
  File "main.py", line 546, in <module>
    main()
  File "main.py", line 536, in main
    train(args, model, processor)
  File "main.py", line 178, in train
    evaluate_knn(args, model, test_features, train_features, processor, set_type="test")
  File "main.py", line 261, in evaluate_knn
    compute_embeddings(model, features, processor, args, verbose=False)
  File "/EE/fewshot_ED/utils.py", line 127, in compute_embeddings
    assert(tok_embeddings.shape[0]==event_feature_num)
AssertionError

使用ACE数据集能正常运行，但是使用MAVEN数据集时出现错误。
错误如上所示，最上面两行打印的是tok_embeddings.shape[0]和event_feature_num值，请问该如何修改，谢谢！

能否公布一下代码？

您好，请问能发布一下模型的代码吗？