1

Google Colab のサイトから simpletransformers の例を実行しようとしています。

例:

from simpletransformers.classification import ClassificationModel, ClassificationArgs
import pandas as pd
import logging


logging.basicConfig(level=logging.INFO)
transformers_logger = logging.getLogger("transformers")
transformers_logger.setLevel(logging.WARNING)

# Preparing train data
train_data = [
    ["Aragorn was the heir of Isildur", 1],
    ["Frodo was the heir of Isildur", 0],
]
train_df = pd.DataFrame(train_data)
train_df.columns = ["text", "labels"]

# Preparing eval data
eval_data = [
    ["Theoden was the king of Rohan", 1],
    ["Merry was the king of Rohan", 0],
]
eval_df = pd.DataFrame(eval_data)
eval_df.columns = ["text", "labels"]

# Optional model configuration
model_args = ClassificationArgs(num_train_epochs=1)

# Create a ClassificationModel
model = ClassificationModel(
    "roberta", "roberta-base", args=model_args
)

# Train the model
model.train_model(train_df)

# Evaluate the model
result, model_outputs, wrong_predictions = model.eval_model(eval_df)

# Make predictions with the model
predictions, raw_outputs = model.predict(["Sam was a Wizard"])

しかし、それは私に次のエラーを与えます:

VersionConflict: このモジュールが正常に機能するには tokenizers==0.9.4 が必要ですが、tokenizers==0.10.0 が見つかりました。試してください: pip installTransformers -U または pip install -e '.[dev]' git master を使用している場合

試してみまし!pip install transformers -U!pip install tokenizers==0.9.4 が、同じエラーが発生し続けます。以前にこのコードを実行したことがありますが、うまくいきましたが、今は前述のエラーが発生しています。

4

2 に答える 2