medicalAI
medicalAI copied to clipboard
Predict function might go into infinite loop on some systems
Solution 1:
Try reinstalling medicalai+ TensorFlow in a new and clean virtual environment.
Solution 2: Use the following modifications
For Prediction:
predictions = infEngine.model.predict(testGen.generator,verbose=1, steps=testGen.STEP_SIZE)
predictions = trainer.model.predict(testGen.generator,verbose=1, steps=testGen.STEP_SIZE)
For generate_evaluation_report:
generate_evaluation_report(testSet=testGen, predictions = infEngine.model.predict( testGen.generator, verbose=1, steps=testGen.STEP_SIZE ) )