Searched refs:seq_length (Results 1 – 4 of 4) sorted by relevance
| /Zephyr-Core-3.5.0/samples/modules/tflite-micro/magic_wand/train/ |
| D | data_load.py | 39 seq_length): argument
41 self.seq_length = seq_length
66 def pad(self, data, seq_length, dim): # pylint: disable=no-self-use argument
71 tmp_data = (np.random.rand(seq_length, dim) - 0.5) * noise_level + data[0]
72 tmp_data[(seq_length -
73 min(len(data), seq_length)):] = data[:min(len(data), seq_length)]
76 tmp_data = (np.random.rand(seq_length, dim) - 0.5) * noise_level + data[-1]
77 tmp_data[:min(len(data), seq_length)] = data[:min(len(data), seq_length)]
85 features = np.zeros((length, self.seq_length, self.dim))
89 padded_data = self.pad(data, self.seq_length, self.dim)
|
| D | train.py | 50 def build_cnn(seq_length): argument
57 input_shape=(seq_length, 3, 1)), # output_shape=(batch, 128, 3, 8)
77 def build_lstm(seq_length): argument
82 input_shape=(seq_length, 3)), # output_shape=(batch, 44)
92 def load_data(train_data_path, valid_data_path, test_data_path, seq_length): argument
94 train_data_path, valid_data_path, test_data_path, seq_length=seq_length)
100 def build_net(args, seq_length): argument
102 model, model_path = build_cnn(seq_length)
104 model, model_path = build_lstm(seq_length)
184 seq_length = 128 variable
[all …]
|
| D | train_test.py | 36 self.seq_length = 128
40 self.seq_length)
48 cnn, cnn_path = build_cnn(self.seq_length)
49 lstm, lstm_path = build_lstm(self.seq_length)
|
| D | data_load_test.py | 34 "./data/train", "./data/valid", "./data/test", seq_length=512)
57 padding_data1 = self.loader.pad(original_data1, seq_length=5, dim=2)
58 padding_data2 = self.loader.pad(original_data2, seq_length=5, dim=2)
|
