重みの可視化に汎用性をもたせる

aist9 / autoencoder

MIT License

1 stars 0 forks source link

Closed uchikun2493 closed 4 years ago

uchikun2493 commented 4 years ago

すでに重みの可視化関数を追加している

encoder_w_plot_for_mnist(
        model, hidden, out_dir=save_dir)
decoder_b_plot_for_mnist(
        model, out_dir=save_dir)

これは現状MNISTにしか対応していないので、１次元データでも対応できるなど汎用性をもたせたい

uchikun2493 commented 4 years ago

以下のブランチを切りました、作業します feature/w_and_b_visual