def main()

in augmentation/augment_bert.py [0:0]

• 13 lines of code
• 1 McCabe index (conditional complexity)

def main(args):
	df = pd.read_csv(args.input_csv, index_col=0)
	kwargs = {
		'top_k': 10,
		'action': 'insert',
		'model_path': args.bert_path,
		'aug_min': 2,
		'aug_max': 4,
		'stopwords': get_stopwords(args.stopwords)
	}
	augmenter = naw.ContextualWordEmbsAug(device=args.device, **kwargs)
	augmented = augment_dataframe(df, augmenter, args.text_key, batch_size=2)
	augmented.to_csv(args.output_csv)