import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense, InputLayer, Dropout, Conv1D, Conv2D, Flatten, Reshape, MaxPooling1D, MaxPooling2D, BatchNormalization, TimeDistributed
from tensorflow.keras.optimizers import Adam
model architecture
model = Sequential()
model.add(Dense(20, activation=‘relu’,
activity_regularizer=tf.keras.regularizers.l1(0.00001)))
model.add(Dense(10, activation=‘relu’,
activity_regularizer=tf.keras.regularizers.l1(0.00001)))
model.add(Dense(classes, activation=‘softmax’, name=‘y_pred’))
this controls the learning rate
opt = Adam(lr=0.0005, beta_1=0.9, beta_2=0.999)
this controls the batch size, or you can manipulate the tf.data.Dataset objects yourself
BATCH_SIZE = 32
train_dataset = train_dataset.batch(BATCH_SIZE, drop_remainder=False)
validation_dataset = validation_dataset.batch(BATCH_SIZE, drop_remainder=False)
callbacks.append(BatchLoggerCallback(BATCH_SIZE, train_sample_count))
train the neural network
model.compile(loss=‘categorical_crossentropy’, optimizer=opt, metrics=[‘accuracy’])
model.fit(train_dataset, epochs=30, validation_data=validation_dataset, verbose=2, callbacks=callbacks)
HI,
I was was to run the project on Edge Impulse. It was running ok. However, when I do it on VSCode, it shows error message. Can you explain why is it and how to fix it? Thanks
NameError: name ‘BatchLoggerCallback’ is not defined
