-
大小: 749KB文件類型: .rar金幣: 2下載: 1 次發布日期: 2021-06-06
- 語言: Python
- 標簽:
資源簡介
基于MovieLens數據集,采用隨機梯度下降算法優化最小化能量函數的概率矩陣分解Python源代碼,自己做實驗的源代碼Probabilistic Matrix Factorization

代碼片段和文件信息
from?numpy?import?*
import?random
def?load_rating_data(file_path=‘ml-100k/u.data‘):
????“““
????load?movie?lens?100k?ratings?from?original?rating?file.
????need?to?download?and?put?rating?data?in?/data?folder?first.
????Source:?http://www.grouplens.org/
????“““
????prefer?=?[]
????for?line?in?open(file_path?‘r‘):??#?打開指定文件
????????(userid?movieid?rating?ts)?=?line.split(‘\t‘)??#?數據集中每行有4項
????????uid?=?int(userid)
????????mid?=?int(movieid)
????????rat?=?float(rating)
????????prefer.append([uid?mid?rat])
????data?=?array(prefer)
????return?data
def?spilt_rating_dat(data?size=0.2):
????train_data?=?[]
????test_data?=?[]
????for?line?in?data:
????????rand?=?random.random()
????????if?rand?????????????test_data.append(line)
????????else:
????????????train_data.append(line)
????train_data?=?array(train_data)
????test_data?=?array(test_data)
????return?train_data?test_data
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????文件??????10089??2017-01-09?23:48??ProbabilisticMatrixFactorization.py
?????文件???????8746??2017-01-12?12:41??Probabilistic-Matrix-Factorization.py
?????文件????1979173??2000-07-19?16:09??data\ml-100k\u.data
?????文件????????958??2017-01-12?12:36??LoadData.py
?????目錄??????????0??2017-01-09?23:19??data\ml-100k
?????目錄??????????0??2017-01-10?20:49??data
-----------?---------??----------?-----??----
??????????????1998966????????????????????6
評論
共有 條評論