sharinkan / HeartDiseasePrediction

0 stars 0 forks source link

Train 2-Dimensional CNN on Mel-Spectrogram Data #21

Closed inari-mike closed 6 months ago

inari-mike commented 6 months ago

A few things have been done:

  1. add 2D-CNN model in cnn2d.py
  2. add new functionalities in preprocessing.py
    • add feature_mel_2d: 2D mel spectrogram feature
    • change build_feature_extractor to make it works for 2D mel spectrogram feature
    • change TCDPdata.getXy, padding time series of mel spectrogram into the same length
    • add high_dim_min_max_scaler and high_dim_standard_scaler. Previous normalizers don't work for 3D or even higher dimensional feature matrices. These 2 can solve the issue.
    • change gen_datesets: introduce high_dim_min_max_scaler and high_dim_standard_scaler to the preprocessing process.
  3. Create a 2-Dimensional CNN training sample notebook, based on Mel-Spectrogram Data: 2dCNN-MelSpec.ipynb
  4. format code in simple MLP training MLP.ipynb
  5. updated requirement.txt