2021-08

Python

【Python】pytorchとBERTで文章分類をする方法

BERTを使えば簡単に文章分類をすることができます。from torch import nnfrom transformers import AutoModelclass Cust...
Python

【Python】LightGBMとtf-idfで自然言語処理モデルを作る方法

自然言語処理の機械学習モデルはBERTが有名ですが、BERTの計算ではGPUが必要です。しかしGPUは高いので、CPUで簡易的に計算したい場面もありますよね。そんな場合はLight...
Python

【Python】pytorchで画像分類モデルを作る方法

pytorchとtimmを使えば、簡単に画像分類モデルを作ることができます。# resnet18dを作るmodel = timm.create_model("resnet18d",...