Toda vez que alguém propõe treinar um modelo para algum domínio (células virtuais/materiais/qualquer coisa), a primeira frase deve ser: dados de entrada: x, dados de saída: y, função de perda: z. Tornaria muito mais fácil para como eu entender o que exatamente está sendo proposto.