天天看點

Librec 獲得訓練集的prediction value

Librec對訓練集和測試集有過濾設定。

資料讀取位址:

https://github.com/guoguibing/librec/blob/3.0.0/core/src/main/java/net/librec/data/splitter/GivenTestSetDataSplitter.java

第98行

// remove test elements from trainMatrix
        for (MatrixEntry me : testMatrix) {
            int rowIdx = me.row();
            int colIdx = me.column();
            trainMatrix.set(rowIdx, colIdx, 0.0);
        }
        trainMatrix.reshape();