initial commit

app.py

@@ -0,0 +1,47 @@
+import numpy as np
+import gradio as gr
+from tensorflow.keras.models import load_model
+import matplotlib
+matplotlib.use('Agg')
+from matplotlib import pyplot as plt
+
+def get_model(path):
+    saved_VRAE_model = load_model(path, compile=False)
+    
+    # return encoder, regressor
+    return saved_VRAE_model.layers[1], saved_VRAE_model.layers[2]
+
+def predict(subset, engine):
+  # ------------------------ Prediction ------------------------
+  path = './models/'+subset+'.h5'
+  y_train = np.load('./data/'+subset+'_y_train.npy', allow_pickle=True)
+  x_test = np.load('./data/'+subset+'_x_test.npy', allow_pickle=True)
+  
+  # Get model
+  encoder, regressor = get_model(path)
+  
+  # latent space in train
+  fig = plt.figure()
+  z_train = np.load('./data/'+subset+'_z_train.npy', allow_pickle=True)
+  plt.scatter(z_train[:, 0], z_train[:, 1], c=y_train, cmap='RdYlGn')
+  clb = plt.colorbar()
+  clb.set_label('RUL', labelpad=-30, y=1.05, rotation=0)
+  z, _, _  = encoder.predict(x_test[engine].reshape(1, x_test[engine].shape[0], x_test[engine].shape[1]))
+  plt.scatter(z[:, 0], z[:, 1], c='red', marker='x')
+  plt.xlabel('z - dim 1')
+  plt.ylabel('z - dim 2')
+  
+  pred = regressor.predict(z)[0][0]
+  
+  return int(pred), fig
+  
+iface = gr.Interface(
+	fn=predict,
+	inputs=[gr.inputs.Radio(["FD001", "FD002", "FD003", "FD004"]), gr.inputs.Slider(1, 100, step=1)],
+	title="RUL Estimation for CMAPSS dataset",
+	description="Enter subset and engine number to predict the remaining useful life (RUL)",
+	outputs=[gr.outputs.Label(label="Prediction"), gr.outputs.Plot(type="auto", label="Latent space")],
+	allow_screenshot=False,
+  theme="dark-peach",
+	layout="unaligned")
+iface.launch()

models/FD001.h5

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

models/FD002.h5

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

models/FD003.h5

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

models/FD004.h5

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

requirements.txt

@@ -0,0 +1,3 @@
+tensorflow==2.3.0
+matplotlib==3.3.4
+pandas==1.1.5