Add the ability to choose visualization

Test plan:

./start_mirt_pipeline.py --sigmoid_viz
./start_mirt_pipeline.py --roc_viz