import torch


def build_classifier_block(input_size: int, number_of_classes: int) -> torch.nn.Sequential:
    """
    Build classifier block designed for obtaining the prediction.

    :param input_size: Input number of features.
    :param number_of_classes: Number of classes.
    :return: Sequential module.
    """
    return torch.nn.Sequential(
        torch.nn.Linear(input_size, 512),
        torch.nn.ReLU(),
        torch.nn.Linear(512, 128),
        torch.nn.ReLU(),
        torch.nn.Linear(128, number_of_classes)
    )