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