This was the exact same problem I was facing.

@Wolf_cola, can you provide pointers on how to go about actually training the model to classify the dataframe with a label? Would a random forest classifier work?

I unfortunately do not have enough credits to comment so needed to write an answer.

Thanks in advance.