diff --git a/Makefile b/Makefile index d5afd87d..1d51fa7e 100644 --- a/Makefile +++ b/Makefile @@ -11,22 +11,22 @@ help: @echo "run Equivalent to eval_onelayer" eval_onelayer: build_onelayer @echo "Running OneLayer Eval on PCAP file $(PCAP)" - docker run -it -v "$(PCAP):/pcaps/eval.pcap" -e SKIP_RABBIT=true poseidonml:onelayer + docker run -it -v "$(PCAP):/pcaps/eval.pcap" -e SKIP_RABBIT=true --entrypoint=python3 poseidonml:onelayer eval_OneLayer.py test_onelayer: build_onelayer @echo "Running OneLayer Test on PCAP files $(PCAP)" - @docker run -it -v "/tmp/models:/OneLayer/models" -v "$(PCAP):/pcaps/" -e SKIP_RABBIT=true poseidonml:onelayer test_OneLayer.py + @docker run -it -v "/tmp/models:/OneLayer/models" -v "$(PCAP):/pcaps/" -e SKIP_RABBIT=true --entrypoint=python3 poseidonml:onelayer train_OneLayer.py train_onelayer: build_onelayer @echo "Running OneLayer Train on PCAP files $(PCAP)" - @docker run -it -v "/tmp/models:/OneLayer/models" -v "$(PCAP):/pcaps/" -e SKIP_RABBIT=true poseidonml:onelayer train_OneLayer.py + @docker run -it -v "/tmp/models:/OneLayer/models" -v "$(PCAP):/pcaps/" -e SKIP_RABBIT=true --entrypoint=python3 poseidonml:onelayer train_OneLayer.py eval_randomforest: build_randomforest @echo "Running RandomForest Eval on PCAP file $(PCAP)" - @docker run -it -v "$(PCAP):/pcaps/eval.pcap" -e SKIP_RABBIT=true poseidonml:randomforest + @docker run -it -v "$(PCAP):/pcaps/eval.pcap" -e SKIP_RABBIT=true --entrypoint=python3 poseidonml:randomforest eval_RandomForest.py test_randomforest: build_randomforest @echo "Running RandomForest Test on PCAP files $(PCAP)" - @docker run -it -v "/tmp/models:/RandomForest/models" -v "$(PCAP):/pcaps/" -e SKIP_RABBIT=true poseidonml:randomforest test_RandomForest.py + @docker run -it -v "/tmp/models:/RandomForest/models" -v "$(PCAP):/pcaps/" -e SKIP_RABBIT=true --entrypoint=python3 poseidonml:randomforest test_RandomForest.py train_randomforest: build_randomforest @echo "Running RandomForest Train on PCAP files $(PCAP)" - @docker run -it -v "/tmp/models:/RandomForest/models" -v "$(PCAP):/pcaps/" -e SKIP_RABBIT=true poseidonml:randomforest train_RandomForest.py + @docker run -it -v "/tmp/models:/RandomForest/models" -v "$(PCAP):/pcaps/" -e SKIP_RABBIT=true --entrypoint=python3 poseidonml:randomforest train_RandomForest.py build_onelayer: build_base @pushd DeviceClassifier/OneLayer && docker build -t poseidonml:onelayer . && popd build_randomforest: build_base