Changes to the algorithm

Zernike_pipeline.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:37ae8f3a37f70e547d8326f272d0bee86c05e81f1bb9c9eb305162dd9d19d18b
-size 4518575
+oid sha256:89b695a5df0e6f06fb59e74839858caebe63519a34dc676c7ed421a8d52cda3b
+size 4584799

app.py

@@ -5,28 +5,26 @@ from joblib import load
 file_name = 'Zernike_pipeline.pkl'
 model = load(file_name)
 
+# cols 6,7,10,13,18,20,23,30
+indices = ['x', 'x', 'x', 'x', 'x', 'x', 'z(0,0)', 'z(1,-1)', 'z(1,+1)', 'z(2,-2)', 'z(2,0)', 'z(2,+2)', 'z(3,-3)', 'z(3,-1)', 'z(3,+1)', 'z(3,+3)', 'z(4,-4)', 'z(4,-2)', 'z(4,0)', 'z(4,+2)', 'z(4,+4)', 'z(5,-5)', 'z(5,-3)', 'z(5,-1)', 'z(5,+1)', 'z(5,+3)', 'z(5,+5)', 'z(6,-6)', 'z(6,-4)', 'z(6,-2)', 'z(6,0)', 'z(6,+2)', 'z(6,+4)', 'z(6,+6)', 'z(7,-7)', 'z(7,-5)', 'z(7,-3)', 'z(7,-1)', 'z(7,+1)', 'z(7,+3)', 'z(7,+5)', 'z(7,+7)']
 
-def Zernike(z1,z2,z3,z4,z5,z6,z7):
+# extract the columns
+z1_label = indices[6]
+z2_label = indices[7]
+z3_label = indices[10]
+z4_label = indices[13]
+z5_label = indices[18]
+z6_label = indices[20]
+z7_label = indices[23]
+z8_label = indices[30]
+
+
+def Zernike(z1,z2,z3,z4,z5,z6,z7,z8):
     print('------------------')
     
-    X = [z1,z2,z3,z4,z5,z6,z7]
+    X = [z1,z2,z3,z4,z5,z6,z7,z8]
     print(X)
     outcome_decoded = ['Normal','Borderline','Suspicious','Keratoconic']
-    file_object = open('zernike_data.txt', 'a')
-    file_object.write(str(z1))
-    file_object.write(';')
-    file_object.write(str(z2))
-    file_object.write(';')
-    file_object.write(str(z3))
-    file_object.write(';')
-    file_object.write(str(z4))
-    file_object.write(';')
-    file_object.write(str(z5))
-    file_object.write(';')
-    file_object.write(str(z6))
-    file_object.write(';')
-    file_object.write('\n')
-    file_object.close()
 
     result = model.predict([X])
     return 'The patient is ' + outcome_decoded[int(result)] + '.'
@@ -34,15 +32,16 @@ def Zernike(z1,z2,z3,z4,z5,z6,z7):
 iface = gr.Interface(
     fn=Zernike, 
     title='Awwad-Versaci Keratoconus Index',
-    description='The Awwad-Versaci Keratoconus Index (AVKI) detects keratoconus and keratoconus susceptible corneas through pachymetry maps centered on the thinnest point and fitted into Zernike polynomials. Beta version by Shady Awwad, Anthony Abou Mrad, MD, Jad Assaf, MD, and Francesco Versaci, PhD. This is the 4-way classification. The Zernke polynomials were taken from the 4mm cornea centered on the thinnest point the selected polynomials are  z(0,0)	z(1,-1)	z(2,0)	z(3,-1)	z(4,0)	z(6,0)	z(7,-1) from highest ROC curves',
+    description='The Awwad-Versaci Keratoconus Index (AVKI) detects keratoconus and keratoconus susceptible corneas through pachymetry maps centered on the thinnest point and fitted into Zernike polynomials. Beta version by Shady Awwad, Anthony Abou Mrad, MD, Jad Assaf, MD, and Francesco Versaci, PhD. This is the 4-way classification. The Zernke polynomials were taken from the 4mm cornea centered on the thinnest point and selected from highest ROC curves',
     inputs=[
-        gr.inputs.Number(label="z(0,0)"),
-        gr.inputs.Number(label="z(1,-1)"),
-        gr.inputs.Number(label="z(2,0)"),
-        gr.inputs.Number(label="z(3,-1)"),
-        gr.inputs.Number(label="z(4,0)"),
-        gr.inputs.Number(label="z(6,0)"),
-        gr.inputs.Number(label="z(7,-1)")
+        gr.inputs.Number(label=z1_label),
+        gr.inputs.Number(label=z2_label),
+        gr.inputs.Number(label=z3_label),
+        gr.inputs.Number(label=z4_label),
+        gr.inputs.Number(label=z5_label),
+        gr.inputs.Number(label=z6_label),
+        gr.inputs.Number(label=z7_label),
+        gr.inputs.Number(label=z8_label)
     ],
     outputs="text")
 iface.launch(