Skip to content

Add Multi-MLP model

choucl requested to merge wip/add-4mlp-model into main

I've added the Multi-MLP model (InceptionTimeMultiMLP). The Multi-MLP model will train 4 different MLPs after extracting features from the IT blocks. image

Changes

  • Add the InceptionTimeMultiMLP class
  • Update training and preprocessing script for the model
  • Subtle change to the preprocessing script to prevent generating unexpected label numbers
  • Subtle change to the single MLP model to reuse the MlpModule class
  • Logging parameter count
Edited by choucl

Merge request reports