Data-driven frameworks for rheology