unravel.classifiers.CrystalGraphClassifier

class unravel.classifiers.CrystalGraphClassifier

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)

Parameters:

• n_layers (int)

• channels (int)

• drop_out (float)

• n_out (int)

Methods

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

__init__(n_layers=3, channels=128, drop_out=0.5, n_out=1, **kwargs)

Parameters:

• n_layers (int)

• channels (int)

• drop_out (float)

• n_out (int)

call(inputs)