added sparsity trainer with quadratic activation and p-norm

saprmarks / dictionary_learning

MIT License

142 stars 37 forks source link

added sparsity trainer with quadratic activation and p-norm #12

Closed rangell closed 4 months ago

rangell commented 6 months ago

Modified AutoEncoder class to take an optional activation function argument. Also implemented SparsityTrainer that takes activation function and fixed p for the l_p sparsity norm.