what does n_class of logits in gumbel_softmax represent?

chaoshangcs / GTS

Discrete Graph Structure Learning for Forecasting Multiple Time Series, ICLR 2021.

Apache License 2.0

171 stars 30 forks source link

Closed turmeric-blend closed 3 years ago

turmeric-blend commented 3 years ago

Hi, what does n_class of the logits input in gumbel_softmax function represent?

chaoshangcs commented 3 years ago

Hi. It represents the number of classes. Here the n_class is 2.

turmeric-blend commented 3 years ago

what does these classes represent? in terms of the dataset.

chaoshangcs commented 3 years ago

It is the general binary classification. Each edge is either 0 or 1.

turmeric-blend commented 3 years ago

I see. Thanks.