AI modelling of animal movements in a heterogeneous habitat