Skip to content

Instantly share code, notes, and snippets.

@um4ng-tiw

um4ng-tiw/Lung cancer detection.py Secret

Created July 13, 2021 06:11
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save um4ng-tiw/a3bc8a99e9d2ce7fbd9be61319596edf to your computer and use it in GitHub Desktop.
Save um4ng-tiw/a3bc8a99e9d2ce7fbd9be61319596edf to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Lung cancer detection.py
x,y = test_data_generator.next()
pred_array=[]
max_index_arr = []
for i in range(5):
img = x[i]
img = img.reshape(-1,64,64,1)
pred_val = model.predict(img)
max_idx = np.argmax(pred_val)
pred_array.append(max_idx)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment