Beispiel #1
0
 def test_show_network_cont_structure(self):
     train = trainner()
     train.load_db(file='DB_MAKER/CNS_predict_abnormal_cont_train_db.bin')
     train.show_network(type_net='GRU', classify=False)
     train.show_network(type_net='LSTM', classify=False)
     train.show_network(type_net='CLSTM', classify=False)
     train.show_network(type_net='RNN', classify=False)
     train.show_network(type_net='CNN', classify=False)
Beispiel #2
0
 def test_train_GRU_clasiffer(self):
     train = trainner()
     train.load_db(
         file='DB_MAKER/CNS_predict_abnormal_accident_train_db.bin')
     train.train_clasiffer(type_net='GRU')
Beispiel #3
0
 def test_train_LSTM_control(self):
     train = trainner()
     train.load_db(file='DB_MAKER/CNS_predict_abnormal_cont_train_db.bin')
     train.train_control(type_net='LSTM')