esa / NIDN

Neural Inverse Design of Nanostructures
GNU General Public License v3.0
38 stars 6 forks source link

Add gumbel-softmax for classification #116

Open gomezzz opened 1 year ago

gomezzz commented 1 year ago

Feature

Desired Behavior / Functionality

An alternative to the currently utilized approach of building a linear combination of materials may be the use of a gumbel-softmax as described here https://arxiv.org/abs/1611.01144 .

What Needs to Be Done

How Can It Be Tested

Run one of the notebooks using the classification with it.