unravel.classifiers.CrystalGraphClassifier

class unravel.classifiers.CrystalGraphClassifier[source]

Default Graph Classifier with CrystalConvolution layers as presented in Sahasrabudhe & Bekkers (2023)

__init__(n_layers=3, channels=128, drop_out=0.5, n_out=1, **kwargs)[source]
Parameters:

Methods

__init__([n_layers, channels, drop_out, n_out])

__init__(n_layers=3, channels=128, drop_out=0.5, n_out=1, **kwargs)[source]
Parameters:
call(inputs)[source]