Data-driven model for the identification of the rock type at a drilling bit