.PHONY: clean-image
clean-image:
	@echo "+ $@"
	@docker rmi cincchallenge  || true

.PHONE: setup
setup:
	mkdir -p model
	mkdir -p result
	
.PHONY: image
image: Dockerfile requirements.txt
	@echo "+ $@"
	@docker build -t cincchallenge -f ./Dockerfile .
	@echo 'Done.'

# EXAMPLE: make train train_data=your_data_directory
.PHONY: train
train: setup image
	@docker run --rm -v $(PWD)/$(train_data):/physionet/training_data -v $(PWD)/model:/physionet/model  -it cincchallenge python train_model.py training_data model 


# EXAMPLE: make train test_data=your_test_data
.PHONY: test
test: setup image
	@docker run --rm -v $(PWD)/model:/physionet/model -v $(PWD)/$(test_data):/physionet/test_data -v $(PWD)/result:/physionet/test_outputs  -it cincchallenge python driver.py model test_data test_outputs

# EXAMPLE: make train test_data=your_test_data train_data=your_data_directory
.PHONY: train_test
train_test: 
	train
	test

.PHONY: clean
clean:
	rm -rf ./result/*
	rm -rf ./model/*