diff --git a/seq_lab.py b/seq_lab.py index cff21d9..498bb1c 100644 --- a/seq_lab.py +++ b/seq_lab.py @@ -54,6 +54,9 @@ vocab = build_vocab(tokens) train_labels = labels_process(tags) train_tokens_ids = data_process(tokens) +print(train_labels[0][:10]) +print(train_tokens_ids[0][:10]) + ner_model = NERModel() nn_model = NeuralNetworkModel(len(train_tokens_ids)) criterion = torch.nn.CrossEntropyLoss()