Function build_classifier 

pub fn build_classifier(
    arch: ModelArchitecture,
    graph: &mut Graph,
    num_classes: usize,
) -> Result<SequentialModel, ModelError>

Builds a full classifier with a custom number of output classes.