Skip to content

Latest commit

 

History

History
37 lines (32 loc) · 1.02 KB

DilatedCNN.md

File metadata and controls

37 lines (32 loc) · 1.02 KB
Raw

An example with the DilatedCNN model

This uses the same example data as the LSTM model, so please generate the data (all the steps until using LSTMModel.py) according to the example in the main README.md first.

Note, resuming training is currently bricked for the DilatedCNNModel.py

training

python3 helixer/prediction/DilatedCNNModel.py \
  --data-dir example/train/ \
  --save-model-path example/best_helixer_model.h5 \
  --epochs 5

predicting

python3 helixer/prediction/DilatedCNNModel.py  \
  --load-model-path example/best_helixer_model.h5 \
  --test-data example/test/test_data.h5 \
  --prediction-output-path example/mpusillaCCMP1545_predictions.h5

eval

python3 helixer/prediction/DilatedCNNModel.py  \
  --load-model-path example/best_helixer_model.h5 \
  --test-data example/test/test_data.h5 --eval

visualization

(have not actually tested)

python3 helixer/visualization/visualize.py --predictions example/mpusillaCCMP1545_predictions.h5 --test-data example/test/test_data.h5