AutoEncoderを使った異常検知を試してみた

テーブルデータの異常検知を学ぶため、下記のサイトのコードを試してみました。 データセットは kaggle の心電図データを使っています。 PTB診断ECGデータベースの正常データをAutoEncoderで学習し、正常データと異常データの異常スコアを算定して比較しました。 データセットの詳細はこちらをご覧ください。 Colab上で実行したコードはこちら。 データセットは Google Drive にアップロードして実行しています。 なお、ニューラルネット詳しくないマンなので、変なところがありましたら優しくご指摘いただけると喜びます。 参考サイトからのコードの変更点 コードは参考サイトから流用しているので詳細は参考サイト(再掲)か、Colab のデータをご覧ください。 なお、使用したデータセットのカラム数 188 に合わせてネットワークの層の数を変更し、1つ層を増やしています。 class A