Jupyter notebook에서 argparse 이용하기

Github에 올라와 있는 다른 사람들의 딥러닝 코드들을 보면 cmd 창에서 변수를 받아 실행하기 위해 argparse 코드를 많이 이용한다.
Jupyter notebook에서 코드가 그대로 실행이 되지 않으므로 약간의 코드 수정을 해야 한다. 매우 간단하다.

다음은 내가 추천시스템 프로젝트를 했을 때의 코드의 일부이다.

Example

parser = argparse.ArgumentParser()
parser.add_argument('--dataset', default='Video', type=str)
parser.add_argument('--train_dir', default='train', type=str)
parser.add_argument('--batch_size', default=128, type=int)
parser.add_argument('--lr', default=0.001, type=float)
parser.add_argument('--maxlen', default=50, type=int)
parser.add_argument('--hidden_units', default=50, type=int)
parser.add_argument('--num_blocks', default=2, type=int)
parser.add_argument('--num_epochs', default=201, type=int)
parser.add_argument('--num_heads', default=1, type=int)
parser.add_argument('--dropout_rate', default=0.5, type=float)
parser.add_argument('--l2_emb', default=0.0, type=float)
args = parser.parse_args()

마지막줄을

args = parser.parse_args()

다음처럼 바꿔주면 된다.

args = parser.parse_args(args=[])

끝