基于Gemma的数据科学Tutorial

Gemma简介

        Gemma是谷歌最近开放的一个轻量级LLM模型，和先前开源的Gemini采用了相同的技术，但目前只支持英文Prompt作为输入。

       因为最近Kaggle上发布了一个任务Google – AI Assistants for Data Tasks with Gemma，发现结合Gemma+python可以很好的对数据进行处理、分析、挖掘、建模等一系列操作，可以帮助很多写代码比较困难的小伙伴。

Tutorial

      本教程基于Kaggle进行，所以没有Kaggle账号的小伙伴可以先注册一个

• 初始化模型

# 导入依赖
import os
os.environ["KERAS_BACKEND"] = "torch" # Or "jax" or "tensorflow".

import keras
import keras_nlp
import numpy as np

# 初始化Gemma模型
gemma_lm = keras_nlp.models.GemmaCausalLM.from_preset("gemma_instruct_2b_en")