圖(graph)神經(jīng)網(wǎng)絡(luò)學(xué)習(xí)(四)--代碼解析(Model_2)

【2】Model_2: 1stChebNet(2017)-github:gcn? (https://github.com/tkipf/gcn)? (tensorflow)

{Semi-Supervised Classification with Graph Convolutional Networks}基于圖卷積網(wǎng)絡(luò)的半監(jiān)督分類

文章:https://www.colabug.com/5231014.html

Graph Convolutional Networks

這是一個用tensorflow框架實現(xiàn)的圖卷積神經(jīng)網(wǎng)絡(luò),用來完成圖中的節(jié)點(diǎn)類型分類废酷,即是一個分類模型匆背。

兩篇論文:

【1】Semi-Supervised Classification with Graph Convolutional Networks?(ICLR 2017)

【2】Graph Convolutional Networks?(2016)



從github下載代碼场晶,上傳到服務(wù)器上(盡量保證你的服務(wù)能連接外網(wǎng)粗卜,因為在調(diào)試程序時需要從外網(wǎng)中下載所需的數(shù)據(jù)庶橱,盡量保證你的服務(wù)器有GPU轻专,因為有的代碼常常會用GPU來加速)蚓峦,解壓:unzip gcn-master.zip

查看虛擬環(huán)境名:conda?env list? ? ?

選擇一個虛擬環(huán)境激活:source activate your_env_name

查看python版本:python? ?(python3)

安裝:python setup.py install

成功安裝

查看requirements??

·tensorflow(>0.12)??

·networkx

運(yùn)行模型:

cd gcn

python train.py? ? ? ?或者自定數(shù)據(jù)集運(yùn)行? ?python train.py --dataset citeseer

正常運(yùn)行結(jié)果:



現(xiàn)在開始在本地pycharm中分析源代碼

setup.py可以看到作者Thomas Kipf已經(jīng)他寫的圖卷積網(wǎng)絡(luò)gcn做成了安裝包徽级,剛才運(yùn)行的python setup.py install就是將該gcn模型安裝到python的lib目錄下气破。

運(yùn)行模型命令: python train.py

以下是train.py代碼,我們主要關(guān)心模型的輸入數(shù)據(jù)格式餐抢。

train.py代碼

以下截圖是train.py的代碼现使,從中可以看出,數(shù)據(jù)集有三個可選旷痕,模型種類有三個可選碳锈。

數(shù)據(jù)集:'cora','citeseer','pubmed'

模型:'gcn','gcn_cheby','dense'

train.py代碼

程序中的數(shù)據(jù)集,有三個整理好的數(shù)據(jù)集:'cora','citeseer','pubmed'

utils.py代碼
utils.py代碼

我們自己的數(shù)據(jù)集欺抗,處理成這個模型可用的格式殴胧。

·N*N的鄰接矩陣? (N是圖節(jié)點(diǎn)的個數(shù))

·N*D的特征矩陣? (D是節(jié)點(diǎn)特征向量的維度)

·N*E?的分類標(biāo)簽矩陣? (E是類別個數(shù))

詳細(xì)理解一下數(shù)據(jù):

ind.dataset_str.x文件?是train訓(xùn)練集實例的特征向量,按scipy.sparse.csr.csr_matrix稀疏矩陣存儲的。

ind.dataset_str.tx文件?是test測試集實例的特征向量团滥,按scipy.sparse.csr.csr_matrix稀疏矩陣存儲的竿屹。

ind.dataset_str.allx文件?是所有(有標(biāo)簽和無標(biāo)簽)的train訓(xùn)練實例特征向量,按scipy.sparse.csr.csr_matrix稀疏矩陣存儲的灸姊。? (半監(jiān)督方法拱燃,一些標(biāo)簽數(shù)據(jù),一些無標(biāo)簽數(shù)據(jù))

ind.dataset_str.y文件?是train訓(xùn)練數(shù)據(jù)集的one-hot類型的標(biāo)簽向量力惯,按numpy.ndarray形式存儲的宋梧。

ind.dataset_str.ty文件?是test測試數(shù)據(jù)集的one-hot類型的標(biāo)簽向量,按numpy.ndarray形式存儲的鹤耍。

ind.dataset_str.ally文件?是所有有標(biāo)簽數(shù)據(jù)的one-hot類型標(biāo)簽向量赞警,按numpy.ndarray形式存儲的。

ind.dataset_str.graph文件?是一個字典{index:[index_of_neighbor_nodes]}

ind.dataset_str.test.index文件?是測試數(shù)據(jù)集的index

以上數(shù)據(jù)都用pickle存儲甲喝。

讀取pickle數(shù)據(jù)顯示如下:

ind.cora.x數(shù)據(jù)? ?train訓(xùn)練數(shù)據(jù)
ind.cora.y數(shù)據(jù)? ?train數(shù)據(jù)集的標(biāo)簽one-hot向量
ind.cora.tx數(shù)據(jù)? ? test測試數(shù)據(jù)
ind.cora.ty數(shù)據(jù)? ?test數(shù)據(jù)集的標(biāo)簽one-hot向量??
ind.cora.allx數(shù)據(jù)? ? 所有(有標(biāo)簽和無標(biāo)簽)的train訓(xùn)練集?
ind.cora.ally數(shù)據(jù)? ??所有有標(biāo)簽數(shù)據(jù)的one-hot類型標(biāo)簽向量??
ind.cora.graph數(shù)據(jù)? ?字典

再理解load_data()函數(shù)

load_data()函數(shù)? 圖1
load_data()函數(shù)? 圖2
oad_data()函數(shù)? 圖3

GCN的實驗效果

這個框架還支持多個圖形實例(可能具有不同大谐⑽)的批量分類,每個實例具有一個鄰接矩陣埠胖。 最好連接各個特征矩陣并構(gòu)建(稀疏)塊對角矩陣糠溜,其中每個塊對應(yīng)于一個圖實例的鄰接矩陣。 對于池化(在圖形級輸出而不是節(jié)點(diǎn)級輸出的情況下)直撤,最好指定一個簡單的池化矩陣非竿,該矩陣從各自的圖形實例中收集特征,如下所示:

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末谋竖,一起剝皮案震驚了整個濱河市红柱,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌蓖乘,老刑警劉巖锤悄,帶你破解...
    沈念sama閱讀 216,651評論 6 501
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異驱敲,居然都是意外死亡铁蹈,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,468評論 3 392
  • 文/潘曉璐 我一進(jìn)店門众眨,熙熙樓的掌柜王于貴愁眉苦臉地迎上來握牧,“玉大人,你說我怎么就攤上這事娩梨⊙匮” “怎么了?”我有些...
    開封第一講書人閱讀 162,931評論 0 353
  • 文/不壞的土叔 我叫張陵狈定,是天一觀的道長颂龙。 經(jīng)常有香客問我习蓬,道長,這世上最難降的妖魔是什么措嵌? 我笑而不...
    開封第一講書人閱讀 58,218評論 1 292
  • 正文 為了忘掉前任躲叼,我火速辦了婚禮,結(jié)果婚禮上企巢,老公的妹妹穿的比我還像新娘枫慷。我一直安慰自己,他們只是感情好浪规,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,234評論 6 388
  • 文/花漫 我一把揭開白布或听。 她就那樣靜靜地躺著,像睡著了一般笋婿。 火紅的嫁衣襯著肌膚如雪誉裆。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,198評論 1 299
  • 那天缸濒,我揣著相機(jī)與錄音足丢,去河邊找鬼。 笑死绍填,一個胖子當(dāng)著我的面吹牛霎桅,可吹牛的內(nèi)容都是我干的栖疑。 我是一名探鬼主播讨永,決...
    沈念sama閱讀 40,084評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼遇革!你這毒婦竟也來了卿闹?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 38,926評論 0 274
  • 序言:老撾萬榮一對情侶失蹤萝快,失蹤者是張志新(化名)和其女友劉穎锻霎,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體揪漩,經(jīng)...
    沈念sama閱讀 45,341評論 1 311
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡旋恼,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,563評論 2 333
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了奄容。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片冰更。...
    茶點(diǎn)故事閱讀 39,731評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖昂勒,靈堂內(nèi)的尸體忽然破棺而出蜀细,到底是詐尸還是另有隱情,我是刑警寧澤奠衔,帶...
    沈念sama閱讀 35,430評論 5 343
  • 正文 年R本政府宣布谆刨,位于F島的核電站,受9級特大地震影響归斤,放射性物質(zhì)發(fā)生泄漏痊夭。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,036評論 3 326
  • 文/蒙蒙 一脏里、第九天 我趴在偏房一處隱蔽的房頂上張望生兆。 院中可真熱鬧,春花似錦膝宁、人聲如沸鸦难。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,676評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽合蔽。三九已至,卻和暖如春介返,著一層夾襖步出監(jiān)牢的瞬間拴事,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,829評論 1 269
  • 我被黑心中介騙來泰國打工圣蝎, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留刃宵,地道東北人。 一個月前我還...
    沈念sama閱讀 47,743評論 2 368
  • 正文 我出身青樓徘公,卻偏偏與公主長得像牲证,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子关面,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,629評論 2 354