前提・実現したいこと
エラーを解決したいです。
Google Colab
発生している問題・エラーメッセージ
AttributeError Traceback (most recent call last)
<ipython-input-8-e55b309388b6> in <module>()
1 small_train_dataset = 'tf_train_dataset'
2 small_eval_dataset = 'tf_eval_dataset'
----> 3 tf_train_dataset = small_train_dataset.remove_columns(["text"]).with_format("tensorflow")
4 tf_eval_dataset = small_eval_dataset.remove_columns(["text"]).with_format("tensorflow")
AttributeError: 'str' object has no attribute 'remove_columns'
AttributeError: 'str' object has no attribute 'remove_columns'
該当のソースコード
import tensorflow as tf
from transformers import TFAutoModelForSequenceClassification
model = TFAutoModelForSequenceClassification.from_pretrained("bert-base-cased", num_labels=2)
small_train_dataset = 'tf_train_dataset'
small_eval_dataset = 'tf_eval_dataset'
tf_train_dataset = small_train_dataset.remove_columns(["text"]).with_format("tensorflow")
tf_eval_dataset = small_eval_dataset.remove_columns(["text"]).with_format("tensorflow")
回答1件
あなたの回答
tips
プレビュー