1. 程式人生 > >用CNN工具箱對自己的資料集分類

用CNN工具箱對自己的資料集分類

在github上下載了一個Deeplearningtoolbox的工具箱 裡面各種deeplearning 很全 我用其中的CNN對自己的資料集分類  但很麻煩 很多地方都得改動  http://download.csdn.net/detail/wd1603926823/9538560這個工具箱裡的CNN用來對自己的資料集分類就只用改動一兩個地方  比較簡單

只用改動cnn-master下的demo資料夾下的cnnTrain.m和config.m這兩個m檔案即可具體改動如下:config.m中 改動我標註釋的兩個地方 我的資料是1000維的所以我把這裡寫100X10 當然也可以寫50X20等等  我是分2類 所以手寫數字的10我改成了我的2

cnnTrain.m中改動如下:載入自己的資料 一定記得資料data是 維數X個數  我的就是1000X328  還有41那裡一定是被個數整除的數   這樣要改動的地方就完了   可以運行了