def predict(dnn, ds, batch = 32):
    """
    Predict if dataset samples represent similar or dissimilar 
    source code solutions of problems
    Parameters:
    - dnn   -- neural network classifier
    - ds    -- dataset as numpy array of samples
    - batch -- batch size for running dnn
    Returns: numpy array of prdiction probabilities
    """
    return dnn.predict(ds, batch_size = batch)