Dipen Chawla DipenChawla

Enthusiastic about tackling challenging and exciting problems using Machine Learning and NLP.

stefan-it / ner_prediction.py

Created September 19, 2019 10:19

Prediction script for PyTorch-Transformers NER

	import click
	import numpy as np
	import torch

	from collections import namedtuple

	from pytorch_transformers import BertForTokenClassification, BertTokenizer

	from torch.nn import CrossEntropyLoss
	from torch.utils.data import DataLoader, SequentialSampler, TensorDataset