Python scipy.sparse 稀疏矩陣訪問速度測試

最近做一個多標籤的任務,需要對預測的top k index計算precision和nDCG。必要的一步是確定預測的index是否在groundtruth中。groundtruth y_dev 是multi-hot形式,使用scipy中的稀疏矩陣存儲。 判斷i號樣本的p號標籤是否預測正確,只需要取 y_dev[i, p]的值即可(命中爲1.0,不命中則爲0.0),當數據量非常大時,訪問的性能不可忽略
相關文章
相關標籤/搜索