After flattening, we use a multi-layer perceptron (MLP) to compute the scalar output. For the regression task of predicting activity we use this output directly, and for the classification task of predicting whether or not the ligand is active against the pocket, we use a sigmoid to give us the output probability.