| 466 | } |
| 467 | |
| 468 | ModelProto MakeBaseModelProto() { |
| 469 | ModelProto model_proto; |
| 470 | auto *sp1 = model_proto.add_pieces(); |
| 471 | auto *sp2 = model_proto.add_pieces(); |
| 472 | auto *sp3 = model_proto.add_pieces(); |
| 473 | |
| 474 | sp1->set_type(ModelProto::SentencePiece::UNKNOWN); |
| 475 | sp1->set_piece("<unk>"); |
| 476 | sp2->set_type(ModelProto::SentencePiece::CONTROL); |
| 477 | sp2->set_piece("<s>"); |
| 478 | sp3->set_type(ModelProto::SentencePiece::CONTROL); |
| 479 | sp3->set_piece("</s>"); |
| 480 | |
| 481 | return model_proto; |
| 482 | } |
| 483 | |
| 484 | // Returns model protos in parameterized tests. |
| 485 | const std::vector<Model::EncoderVersion> &GetEncoderVersions() { |
no test coverage detected