UTokyo-ICEPP/multiml_htautau

multiml_htautau

multiml_htautau is a code for Event Classification with Multi-step Machine Learning(arxiv) in vCHEP 2021.

Dependency

multiml (https://github.com/UTokyo-ICEPP/multiml)

Installation

$ pip install -e .

Run

Scripts for DARTS is in examples/keras, and scripts for SPOS-NAS is in examples/pytorch.

Grid search with Keras:

$ python run_multi_connection_grid.py --tau4vec_weights 0.5 -n 50000 --data_path DATA_PATH

DARTS with Keras:

$ python run_multi_connection_darts.py --individual_loss_weights 1.0 --tau4vec_weights 0.5 -n 50000 --data_path DATA_PATH

SPOS-NAS with PyTorch:

$ python run_multi_connection_sposnas.py --weight 0.5 -e 50000 --data_path DATA_PATH

Tests and coding style

Please test and apply yapf before you commit changes.

$ python setup.py test

$ yapf -i [changed file]