Inferring genotype-phenotype maps using attention models