Benchmarking machine learning models for quantum state classification