2

次のような単純なトランスフォーマー スクリプトがあります。

from simpletransformers.seq2seq import Seq2SeqModel, Seq2SeqArgs
args = Seq2SeqArgs()
args.num_train_epoch=5
model = Seq2SeqModel(
    "roberta",
    "roberta-base",
    "bert-base-cased",
)
import pandas as pd
df = pd.read_csv('english-french.csv')
df['input_text'] = df['english'].values
df['target_text'] =df['french'].values
model.train_model(df.head(1000))
print(model.eval_model(df.tail(10)))

eval_loss は{'eval_loss': 0.0001931049264385365}

ただし、予測スクリプトを実行すると

to_predict = ["They went to the public swimming pool."]
predictions=model.predict(to_predict)

私はこれを得る

['']

使用したデータセットはこちら

出力について非常に混乱しています。何も返さない理由や説明をいただければ幸いです。

4

1 に答える 1