Skip to content

i4never/bert4keras-CMLM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

bert4keras-CMLM

inputs:

  • token_ids: [batch_size, 2*N+1, max_len] 其中仅第 N 行包含mask
  • segment_ids: [batch_size, 2*N+1, max_len]
  • target_ids: [batch_size, max_len]
  • is_masked: [batch_size, max_len]

outputs:

  • loss: [batch_size, ]
  • acc: [batch_size, ]

outputs实际没有用到,保证第一个维度为batch_size即可

Releases

No releases published

Packages

No packages published

Languages