Learning heuristics for A