# 主要是用索引信息 用監(jiān)督樣本來(lái)做訓(xùn)練
def train(model, epochs):
model.train()
optimizer.zero_grad()
output, att, emb1, com1, com2, emb2, emb= model(features, sadj, fadj)
loss_class = F.nll_loss(output[idx_train], labels[idx_train])
# !!!主要是這里 監(jiān)督損失的索引只用idx_train的!!!
loss_dep = (loss_dependence(emb1, com1, config.n) + loss_dependence(emb2, com2, config.n))/2
loss_com = common_loss(com1,com2)
loss = loss_class + config.beta * loss_dep + config.theta * loss_com
acc = accuracy(output[idx_train], labels[idx_train])
loss.backward()
optimizer.step()
acc_test, macro_f1, emb_test = main_test(model)
print('e:{}'.format(epochs),
'ltr: {:.4f}'.format(loss.item()),
'atr: {:.4f}'.format(acc.item()),
'ate: {:.4f}'.format(acc_test.item()),
'f1te:{:.4f}'.format(macro_f1.item()))
return loss.item(), acc_test.item(), macro_f1.item(), emb_test
def main_test(model):
model.eval()
output, att, emb1, com1, com2, emb2, emb = model(features, sadj, fadj)
acc_test = accuracy(output[idx_test], labels[idx_test])
# !!!這里測(cè)試的時(shí)候就用了測(cè)試的索引!!!
label_max = []
for idx in idx_test:
label_max.append(torch.argmax(output[idx]).item())
labelcpu = labels[idx_test].data.cpu()
macro_f1 = f1_score(labelcpu, label_max, average='macro')
如何用標(biāo)簽信息做半監(jiān)督學(xué)習(xí)
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。
相關(guān)閱讀更多精彩內(nèi)容
- 監(jiān)督學(xué)習(xí)(supervised learning):已知數(shù)據(jù)和其一一對(duì)應(yīng)的標(biāo)簽,訓(xùn)練一個(gè)智能算法,將輸入數(shù)據(jù)映射到...
- 監(jiān)督學(xué)習(xí) 給機(jī)器訓(xùn)練的數(shù)據(jù)集是擁有”標(biāo)記“或者”答案的“ 非監(jiān)督學(xué)習(xí) 給機(jī)器的訓(xùn)練的數(shù)據(jù)沒(méi)有任何“標(biāo)記”或者“答案...
- Machine learning機(jī)器學(xué)習(xí)是Artificial inteligence的核心,分為四類: 1、Su...
- 什么是監(jiān)督學(xué)習(xí)、無(wú)監(jiān)督學(xué)習(xí)、強(qiáng)化學(xué)習(xí)、弱監(jiān)督學(xué)習(xí)、半監(jiān)督學(xué)習(xí)、多示例學(xué)習(xí)? 隨著機(jī)器學(xué)習(xí)問(wèn)題不斷深入人心,人們也將...
- 根據(jù)數(shù)據(jù)類型的不同,對(duì)一個(gè)問(wèn)題的建模有不同的方式。依據(jù)不同的學(xué)習(xí)方式和輸入數(shù)據(jù), 機(jī)器學(xué)習(xí)主要分為以下四種學(xué)習(xí)方式...