The number of tasks for model
Value:
Number of classes to predict.
Number of features per atom.
Dropout probability, applied after each propagation step and gather step.
Number of features for atom in the graph.
Number of features for each molecule.