medicalAI icon indicating copy to clipboard operation
medicalAI copied to clipboard

Predict function might go into infinite loop on some systems

Open aibharata opened this issue 5 years ago • 0 comments

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 ) )

aibharata avatar Jun 06 '20 20:06 aibharata