There are a lot of wrongs in your code, like '--train-set' in argparse, but you use 'train_set' for code. Check again!!