O ShieldGemma é um conjunto de jogos prontos, pondera modelos de classificador de conteúdo, criados no Gemma 2, que podem determinar se o conteúdo fornecido pelo usuário, gerado pelo modelo ou misto viola uma política de segurança de conteúdo. O ShieldGemma é treinado para identificar quatro danos: conteúdo sexual, conteúdo perigoso, assédio e ódio fala e vem em três variantes de tamanho: 2B, 9B e 27B que permitem equilibrar velocidade, desempenho para atender às suas necessidades em qualquer implantação. Consulte o card de modelo para saber mais sobre a diferença entre essas variantes.
Proteja seus modelos com o ShieldGemma
Iniciar o Google Colab (Keras) | Iniciar o Google Colab (Transformers) |
É possível usar modelos do ShieldGemma nos seguintes frameworks.
- KerasNLP (link em inglês), com checkpoints de modelo disponíveis em Kaggle, Confira o ShieldGemma no Keras Colab para começar.
- Hugging Face Transformers, com pontos de verificação de modelos disponíveis do Hugging Face Hub (links em inglês). Consulte o ShieldGemma no Transformers Colab para ter começar.