def __init__(self):
"""
Initializes the layers of the convolutional recurrent neural network.
"""
super().__init__()
self.convlstm = convLSTM.ConvLSTM(3, 10, (3, 3), 6, True, True, False)
self.flat = nn.Flatten()
self.lin1 = nn.Linear(10 * 240 * 320, 512)
self.relu = nn.ReLU()
self.lin2 = nn.Linear(512, 3)
def operate_algorithm(self):
self.algorithm = self.choose_algorithm.currentText()
self.data = self.choose_data.currentText()
if (self.algorithm != "相似模型" and self.data == "训练数据"):
if (self.name == -1):
self.error_output.clear()
self.show_error("请先选择文件，再点击运行")
elif (self.name != 1 and self.name != 2 and self.name != 3
and self.name != 4):
self.error_output.clear()
self.show_error("文件名字有误，无法判断数据形式")
else:
self.error_output.clear()
self.output_widget.clear()
self.show_message("testing...")
lstm = convLSTM.ConvLSTM()
if (self.algorithm[-1] == 'D'):
self.result = lstm.test(self.name, self.algorithm[:-3], 1)
else:
self.result = lstm.test(self.name, self.algorithm, 0)
self.output_widget.clear()
self.show_message(self.result)
if (self.algorithm[-1] == 'D'):
lstm.draw_image(self.name, self.algorithm[:-3], 1)
else:
lstm.draw_image(self.name, self.algorithm, 0)
self.output_figure.setPixmap(
QPixmap("./result_img/FD00" + str(self.name) + "_" +
self.algorithm + ".png"))
elif (self.algorithm != "相似模型" and self.data == "真实数据"):
if (self.name == -1):
self.error_output.clear()
self.show_error("请先选择文件，再点击运行")
elif (self.name != 1 and self.name != 2 and self.name != 3
and self.name != 4):
self.error_output.clear()
self.show_error("文件名字有误，无法判断数据形式")
else:
self.error_output.clear()
self.output_widget.clear()
self.show_message("predicting...")
lstm = convLSTM.ConvLSTM()
if (self.algorithm[-1] == 'D'):
self.result = lstm.real(self.name, self.algorithm[:-3], 1)
else:
self.result = lstm.real(self.name, self.algorithm, 0)
self.output_widget.clear()
self.show_message(self.result)
elif (self.algorithm == "相似模型" and self.data == "训练数据"):
if (self.name == -1):
self.error_output.clear()
self.show_error("请先选择文件，再点击运行")
elif (self.name != 1 and self.name != 2 and self.name != 3
and self.name != 4):
self.error_output.clear()
self.show_error("文件名字有误，无法判断数据形式")
elif (self.name == 3 or self.name == 4):
self.error_output.clear()
self.show_error("该数据集不适合用相似匹配法进行预测！")
else:
self.error_output.clear()
self.output_widget.clear()
self.show_message("testing...")
# similar = similarityRUL.similarity()
if (self.name == 1):
n_flights = 100
else:
n_flights = 259
test_set = self.path
RUL_set = "./datafolder/RUL_FD00" + str(self.name) + ".txt"
# self.result = similar.test(n_flights, test_set, RUL_set)
self.SimilarTrain = SimilarTrainThread(n_flights, test_set,
RUL_set)
self.SimilarTrain.result_signal.connect(
self.SimilarTrainOutput)
self.SimilarTrain.start()
# print(type(self.result))
elif (self.algorithm == "相似模型" and self.data == "真实数据"):
if (self.name == -1):
self.error_output.clear()
self.show_error("请先选择文件，再点击运行")
elif (self.name != 1 and self.name != 2 and self.name != 3
and self.name != 4):
self.error_output.clear()
self.show_error("文件名字有误，无法判断数据形式")
elif (self.name == 3 or self.name == 4):
self.error_output.clear()
self.show_error("该数据集不适合用相似匹配法进行预测！")
else:
self.error_output.clear()
self.output_widget.clear()
self.show_message("predicting...")
similar = similarityRUL.similarity()
if (self.name == 1):
n_flights = 100
else:
n_flights = 259
real_set = self.path
self.result = similar.real(n_flights, real_set)
self.output_widget.clear()
self.show_message(self.result)
else:
self.error_output.clear()
self.show_error("选择错误")
def operate_algorithm(self):
self.algorithm = self.choose_algorithm.currentText()
self.data = self.choose_data.currentText()
if (self.algorithm == "LSTM" and self.data == "训练数据"):
if (self.name == -1):
self.error_output.clear()
self.show_error("请先选择文件，再点击运行")
elif (self.name != 1 and self.name != 2 and self.name != 3
and self.name != 4):
self.error_output.clear()
self.show_error("文件名字有误，无法判断数据形式")
else:
self.error_output.clear()
self.output_widget.clear()
self.show_message("testing...")
lstm = convLSTM.ConvLSTM()
self.result = lstm.test(self.name, 'JANET', 0)
self.output_widget.clear()
self.show_message(self.result)
lstm.draw_image(self.name, 'JANET', 0)
elif (self.algorithm == "LSTM" and self.data == "真实数据"):
if (self.name == -1):
self.error_output.clear()
self.show_error("请先选择文件，再点击运行")
elif (self.name != 1 and self.name != 2 and self.name != 3
and self.name != 4):
self.error_output.clear()
self.show_error("文件名字有误，无法判断数据形式")
else:
self.error_output.clear()
self.output_widget.clear()
self.show_message("predicting...")
lstm = convLSTM.ConvLSTM()
self.result = lstm.real(self.name, 'JANET', 0)
self.output_widget.clear()
self.show_message(self.result)
elif (self.algorithm == "相似模型" and self.data == "训练数据"):
if (self.name == -1):
self.error_output.clear()
self.show_error("请先选择文件，再点击运行")
elif (self.name != 1 and self.name != 2 and self.name != 3
and self.name != 4):
self.error_output.clear()
self.show_error("文件名字有误，无法判断数据形式")
elif (self.name == 3 or self.name == 4):
self.error_output.clear()
self.show_error("该数据集不适合用相似匹配法进行预测！")
else:
self.error_output.clear()
self.output_widget.clear()
self.show_message("testing...")
similar = similarityRUL.similarity()
if (self.name == 1):
n_flights = 100
else:
n_flights = 259
test_set = self.path
RUL_set = "./datafolder/RUL_FD00" + str(self.name) + ".txt"
self.result = similar.test(n_flights, test_set, RUL_set)
self.output_widget.clear()
self.show_message(self.result)
elif (self.algorithm == "相似模型" and self.data == "真实数据"):
if (self.name == -1):
self.error_output.clear()
self.show_error("请先选择文件，再点击运行")
elif (self.name != 1 and self.name != 2 and self.name != 3
and self.name != 4):
self.error_output.clear()
self.show_error("文件名字有误，无法判断数据形式")
elif (self.name == 3 or self.name == 4):
self.error_output.clear()
self.show_error("该数据集不适合用相似匹配法进行预测！")
else:
self.error_output.clear()
self.output_widget.clear()
self.show_message("predicting...")
similar = similarityRUL.similarity()
if (self.name == 1):
n_flights = 100
else:
n_flights = 259
real_set = self.path
self.result = similar.real(n_flights, real_set)
self.output_widget.clear()
self.show_message(self.result)
else:
self.error_output.clear()
self.show_error("选择错误")