KerasNLP

  KerasNLP是构建在Keras3.0之上的NLP工具,该软件包支持用户构建整个NLP流程,借助于Keras3.0,使得一套代码同时运行在TF、Pytorch、Jax多个后端成为可能。

  安装方式直接通过pip的方式安装,但是因为Keras3.0刚发布,安装需要一些小技巧

pip install —upgrade keras-nlp
pip install —upgrade keras>=3

Keras3.0需要后安装,不然会导致基于TF的后端覆盖掉最新版本的Keras。

下面是使用KerasNLP基于Bert进行分类的例子:

import os
os.environ["KERAS_BACKEND"] = "tensorflow"  # Or "jax" or "torch"!

import keras_nlp
import tensorflow_datasets as tfds

imdb_train, imdb_test = tfds.load(
    "imdb_reviews",
    split=["train", "test"],
    as_supervised=True,
    batch_size=16,
)
# Load a BERT model.
classifier = keras_nlp.models.BertClassifier.from_preset(
    "bert_base_en_uncased", 
    num_classes=2,
)
# Fine-tune on IMDb movie reviews.
classifier.fit(imdb_train, validation_data=imdb_test)
# Predict two new examples.
classifier.predict(["What an amazing movie!", "A total waste of my time."])

参考:https://keras.io/keras_nlp/

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容