# 载入训练好的模型
import re
import time
import jieba
import jieba.analyse
import jieba.posseg as pseg
import torch
from transformers import BertTokenizer, BertForSequenceClassification, AutoConfig
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") # 调用GPU
"""
your model
"""
class YourModel(object):
def __init__(self,model_path=your_model_path): #此项为初始化项,可有可没有
self.model_path= model_path
print("init model")
self.model=self.init_model()
print("init model done")
def init_model(self):
'''your code, allow return'''
if __name__ == "__main__": #主函数
model=YourModel()
starttime = time.time()
res=model.init_model()
print(res)
endtime = time.time()
print ("times:"+str((endtime - starttime)))
python 常规class类写法及应用
最新推荐文章于 2024-06-13 23:58:10 发布