Projekat iz predmeta SOFT kompjuting.
U pitanju je konvolutivna neuronska mreza, koja klasifikuje rendgenske snimke pluća na tri kategorije:
- Bakterijsku pneumoniu
- Virusnu pneumoniu
- Normalan nalaz
Skup podataka je moguće pronaći ovde. Neophodno ga je otpakovati
i staviti u root direktorijuma.
Program se može pokrenuti sa nekoliko argumenata komandne linije. Svi argumenti su opcioni i imaju podrazumevane vrednosti.
--load_model_from_file=False ili --load_model_from_file=True
Određuje da li da se model učita iz fajla model.dat.
Default: False
--num_epochs=num
Specifiše broj epoha. Default: 5
--run_test=False ili --run_test=True
Određuje da li da se postojeći model evaluira na test skupu. Default: False
--runs_on_cloud=False ili --runs_on_cloud=True
Prilagođava način pokretanja Googlovom ml enginu,
koristiti samo ako se pokreće u tom okruženju. Default: False