first commit

flight_crash_fatality_prediction.py

@@ -0,0 +1,73 @@
+# script to create a new (normalized) observation data point for the plane crash dataset
+# author tbartolo
+import joblib
+import numpy as np
+import gradio as gr
+import os
+
+resources_dir = "resources"
+normalizers_dir = "normalizers"
+algorithms_dir = "algorithms"
+
+
+def master_fn(year, operator, location, ac_type, passengers, crew, algorithm):
+    def ac_type_normalizer(value):
+        hasher = joblib.load(os.path.join(resources_dir, normalizers_dir, "ac_type_hash_model.joblib"))
+        return hasher.transform([[value]]).toarray().flatten()
+
+    def location_normalizer(value):
+        hasher = joblib.load(os.path.join(resources_dir, normalizers_dir, "location_hash_model.joblib"))
+        return hasher.transform([[value]]).toarray().flatten()
+
+    def operator_normalizer(value):
+        hasher = joblib.load(os.path.join(resources_dir, normalizers_dir, "operator_hash_model.joblib"))
+        return hasher.transform([[value]]).toarray().flatten()
+
+    def year_normalizer(value):
+        kbins_discretizer = joblib.load(os.path.join(resources_dir, normalizers_dir, "year_qbins_model.joblib"))
+        return kbins_discretizer.transform([[value]]).flatten()
+
+    def passenger_crew_normalizer(value1, value2):
+        scaler = joblib.load(os.path.join(resources_dir, normalizers_dir, "passenger_crew_scaler_model.joblib"))
+        return scaler.transform([[value1, value2]])[0]
+
+    observation = np.concatenate((year_normalizer(year), operator_normalizer(operator), location_normalizer(location),
+                                  ac_type_normalizer(ac_type), passenger_crew_normalizer(passengers, crew)))
+
+    result = float
+
+    if algorithm == "SVR":
+        svr = joblib.load(os.path.join(resources_dir, algorithms_dir, "svr_model.joblib"))
+        result = svr.predict([observation])[0]
+    elif algorithm == "KNN":
+        knn = joblib.load(os.path.join(resources_dir, algorithms_dir, "knn_model.joblib"))
+        result = knn.predict([observation])[0]
+    elif algorithm == "Isolation Forest":
+        random_forest = joblib.load(os.path.join(resources_dir, algorithms_dir, "random_forest_model.joblib"))
+        result = random_forest.predict([observation])[0]
+    else:
+        raise ValueError('Unknown algorithm: {}'.format(algorithm))
+
+    return round(result, 2)
+
+
+demo = gr.Interface(
+    master_fn,
+    [
+        "number",
+        "text",
+        "text",
+        "text",
+        "number",
+        "number",
+        gr.Radio(["KNN", "Isolation Forest", "SVR"])
+    ],
+    "number",
+    title="Flight Crash Fatality Prediction",
+    description="Enter the flight details and the algorithm to perform the prediction. Note that the purpose of this "
+                "effort is strictly to showcase the strengths and weaknesses amongst the 3 techniques used, "
+                "and by no means claims any sort of accuracy in its predictions.",
+)
+
+if __name__ == "__main__":
+    demo.launch(show_api=False)

resources/algorithms/knn_model.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:113ac752538bdfab02239dfe94373efb6e3422263b10dc7346fdcb0c79fa343b
+size 558836

resources/algorithms/random_forest_model.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b4e419bf578ecedbafe150d168456ff2a0eb7a384730930b45599004d3270250
+size 612001

resources/algorithms/svr_model.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6acfde665a77f17366a12c1131d4b7e2578de717305e54fffef20892b4980654
+size 433620

resources/normalizers/ac_type_hash_model.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4ebc099e2768f32ce9034b6857a1f5cc52de67ca9c5681c3813ea3fb8e915c70
+size 183

resources/normalizers/location_hash_model.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4ebc099e2768f32ce9034b6857a1f5cc52de67ca9c5681c3813ea3fb8e915c70
+size 183

resources/normalizers/operator_hash_model.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4ebc099e2768f32ce9034b6857a1f5cc52de67ca9c5681c3813ea3fb8e915c70
+size 183

resources/normalizers/passenger_crew_scaler_model.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ca4532ad1463f26ac6e0e5cbd6d6ab5bcf0088f94a8ccb13367f61ecef9c5915
+size 1063

resources/normalizers/year_qbins_model.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8ccc8e0cf73b8844100abe773f96a63598a9bb158095743530d57972ff03bdef
+size 1599