DIA
/
enose_2025
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

again

main
ifiguero 2025-03-16 22:36:53 -03:00
parent c8439445f0
commit 99ae1569e3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TrainerClass.py
View File

@ -191,9 +191,9 @@ class eNoseTrainer:
# maxpool 2 o 3 => 2 | 2/15
def build_model_conv1D(config, input_shape, output_dim):
model = keras.Sequential([
layers.Conv1D(filters=config['filters'], kernel_size=config['kernel_l1'], stride=config['kernel_l1']//2, activation='relu', padding='causal', input_shape=input_shape),
layers.Conv1D(filters=config['filters'], kernel_size=config['kernel_l1'], strides=config['kernel_l1']//2, activation='relu', padding='causal', input_shape=input_shape),
layers.MaxPooling1D(pool_size=config['pool_size']),
layers.Conv1D(filters=config['filters'] * 2, kernel_size=config['kernel_l2'], stride=config['kernel_l2']//2, activation='relu', padding='causal'),
layers.Conv1D(filters=config['filters'] * 2, kernel_size=config['kernel_l2'], strides=config['kernel_l2']//2, activation='relu', padding='causal'),
layers.MaxPooling1D(pool_size=config['pool_size']),
layers.Flatten(),
layers.Dense(config['dense_units'], activation='relu'),