Sensorless Drive Diagnosis Feature Classifier - NN Classifier

Edge Impulse Inc. / Sensorless Drive Diagnosis Feature Classifier Public

Clone this project

OmarTests

Primary version

Target: Cortex-M4F 80MHz

Neural Network settings

Training settings

Number of training cycles

Please provide a valid number of training cycles (numeric only)

Learning rate

Please provide a valid number for the learning rate (between 0 and 1)

Advanced training settings

Validation set size

Please provide a valid number for the train/validate split (between 0 and 1)

Split train/validation set on metadata key

Batch size

Auto-balance dataset

Profile int8 model

Neural network architecture

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(90, activation='tanh', activity_regularizer=tf.keras.regularizers.l1(0.00001))) model.add(Dropout(0.25)) model.add(Dense(120, activation='tanh', activity_regularizer=tf.keras.regularizers.l1(0.00001))) model.add(Dropout(0.25)) model.add(Dense(90, activation='tanh', activity_regularizer=tf.keras.regularizers.l1(0.00001))) model.add(Dropout(0.25)) model.add(Dense(12, activation='tanh', 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)) callbacks = tf.keras.callbacks.EarlyStopping(monitor='val_loss', patience=30) # train the neural network model.compile(loss='categorical_crossentropy', optimizer=opt, metrics=['accuracy']) model.fit(train_dataset, epochs=600, validation_data=validation_dataset, verbose=2, callbacks=callbacks)

Input layer (48 features)

Dense layer (90 neurons)

Dense layer (120 neurons)

Dense layer (90 neurons)

Dense layer (12 neurons)

Output layer (11 classes)

Model

Model version:

Did you know? You have access to the full set of Keras layers through the Expert view (click on to switch),
or can even bring your own model (in PyTorch, Keras or scikit-learn).

Layer type
Dense Fully connected layer, the simplest form of a neural network layer. Use this for processed data, such as the output of a spectral analysis DSP block.
1D Convolution / pooling Learn features that take spatial information into account along a single dimension. Use this for raw data, or for DSP blocks that output spatial data, such as the MFCC block.
2D Convolution / pooling Learn features that take spatial information into account along two dimensions. Use this for raw data, or for DSP blocks that output spatial data, such as the MFCC block.
Reshape Turn one-dimensional data from a DSP block into multi-dimensional data. Use this as an input to a convolutional layer. Use this for deep learning on raw data, or to process MFCC output.
Flatten Flatten multi-dimensional data into a single dimension. You need to flatten data from a convolutional layer before returning.
Dropout Reduce the risk of a model overfitting your dataset by randomly cutting a fraction of network connections during training. Can be helpful if your model's training performance is better than its validation performance.

Complete architectures

Model	Author
PyTorch MLP example (20x10 hidden layers) Enterprise Custom ML block with PyTorch	Edge Impulse Inc.
MCSA Enterprise A learning block designed specifically for the MCSA project.	Edge Impulse Inc.
Linear regression (scikit-learn) Enterprise Implementation of sklearn.linear_model.LogisticRegression	Edge Impulse Inc.
LGBM Random Forest Enterprise LGBM Random Forest	Edge Impulse Inc.
LGBM Binary Classification Enterprise LGBM Binary Classification	Edge Impulse Inc.
[unpublished] LGBM Random Forest Classifier Enterprise [unpublished] LGBM Random Forest Classifier	Edge Impulse Inc.
[unpublished] XGBoost Random Forest Classifier Enterprise [unpublished] XGBoost Random Forest Classifier	Edge Impulse Inc.
[unpublished] SVM Classifier Enterprise [unpublished] SVM Classifier	Edge Impulse Inc.

These users will be notified over email when this job finishes.

	Name
	DS David Schwarz Edge Impulse staff

Neural Network settings

Training settings

Advanced training settings

Neural network architecture

Model

Dense

1D Convolution / pooling

2D Convolution / pooling

Reshape

Flatten

Dropout

PyTorch MLP example (20x10 hidden layers)
Enterprise

MCSA
Enterprise

Linear regression (scikit-learn)
Enterprise

LGBM Random Forest
Enterprise

LGBM Binary Classification
Enterprise

[unpublished] LGBM Random Forest Classifier
Enterprise

[unpublished] XGBoost Random Forest Classifier
Enterprise

[unpublished] SVM Classifier
Enterprise

Description

Last training performance (validation set)

Accuracy

Loss

Confusion matrix (validation set)

Feature explorer (full training set)

Developer

Enterprise

Public

Private

( of remaining)

Select a version

Neural Network settings

Training settings

Advanced training settings

Neural network architecture

Model

Description

Last training performance (validation set)

Accuracy

Loss

Confusion matrix (validation set)

Feature explorer (full training set)

Settings