Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/bojone/r-drop
/ functions
Functions
30 in github.com/bojone/r-drop
⨍
Functions
30
◇
Types & classes
11
↓ 5 callers
Method
predict
(self, inputs, output_ids, states)
csl.py:113
↓ 3 callers
Function
load_data
加载数据 单条格式:(文本, 标签id)
sentiment_ssl.py:35
↓ 3 callers
Function
load_data
加载数据 单条格式:(标题, 正文)
csl.py:33
↓ 2 callers
Function
evaluate
(data)
sentiment_ssl.py:144
↓ 2 callers
Function
load_data
(filename)
tnews.py:31
↓ 2 callers
Function
load_data
加载数据 单条格式:(文本, 标签id)
iflytek.py:27
↓ 1 callers
Function
evaluate
(data)
tnews.py:112
↓ 1 callers
Function
evaluate
(data)
iflytek.py:111
↓ 1 callers
Method
evaluate
(self, data, topk=1)
csl.py:146
↓ 1 callers
Method
generate
(self, text, topk=1)
csl.py:119
Method
__init__
(self)
tnews.py:125
Method
__init__
(self)
sentiment_ssl.py:157
Method
__init__
(self)
csl.py:133
Method
__init__
(self)
iflytek.py:124
Method
__iter__
(self, random=False)
tnews.py:56
Method
__iter__
(self, random=False)
sentiment_ssl.py:64
Method
__iter__
(self, random=False)
sentiment_ssl.py:82
Method
__iter__
(self, random=False)
csl.py:62
Method
__iter__
(self, random=False)
iflytek.py:55
Method
compute_loss
(self, inputs, mask=None)
csl.py:81
Function
crossentropy_with_rdrop
配合R-Drop的交叉熵损失
tnews.py:95
Function
crossentropy_with_rdrop
配合R-Drop的交叉熵损失
iflytek.py:94
Function
kld_rdrop
无监督部分只需训练KL散度项
sentiment_ssl.py:129
Method
on_batch_end
(self, batch, logs=None)
sentiment_ssl.py:172
Method
on_epoch_end
(self, epoch, logs=None)
tnews.py:128
Method
on_epoch_end
(self, epoch, logs=None)
sentiment_ssl.py:161
Method
on_epoch_end
(self, epoch, logs=None)
csl.py:138
Method
on_epoch_end
(self, epoch, logs=None)
iflytek.py:127
Function
predict_to_file
输出预测结果到文件 结果文件可以提交到 https://www.cluebenchmarks.com 评测。
tnews.py:139
Function
predict_to_file
输出预测结果到文件 结果文件可以提交到 https://www.cluebenchmarks.com 评测。
iflytek.py:138