How to modify my model with adding some conditions?

Question

I have written this code:

def my_alg(df_train, df_test):
     
    df_train = df_train.to_numpy()
    df_test = df_test.to_numpy()
    
# our metrics
    tp =