Skip to content

Commit 2532e0b

Browse files
committed
* Add implicit feedback methods to package init file.
1 parent d34c25a commit 2532e0b

File tree

4 files changed

+11
-9
lines changed

4 files changed

+11
-9
lines changed

Diff for: .gitignore

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ cmake_install.cmake
44
Makefile
55
build/
66
dist/
7-
pypackage/
7+
pypackage/setup.py
88
pyreclab.egg-info/
99
pyreclab/
1010
install_manifest.txt

Diff for: examples/testIFAls.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
if __name__ == '__main__':
55

6-
ifals = pyreclab.IFAls( dataset = 'lastfm-dataset.txt',
6+
ifals = pyreclab.IFAls( dataset = 'dataset/u1.base',
77
dlmchar = b'\t',
88
header = False,
99
usercol = 0,
@@ -16,8 +16,8 @@
1616
end = time.clock()
1717
print( 'training time: ' + str( end - start ) )
1818

19-
ranking = ifals.recommend( 'user_000436', 5, includeRated = False )
20-
print( 'recommendation for user user_000436: ' + str( ranking ) )
19+
ranking = ifals.recommend( '457', 5, includeRated = False )
20+
print( 'recommendation for user 457: ' + str( ranking ) )
2121

2222
print( '-> recommendation test' )
2323
start = time.clock()

Diff for: examples/testIFAlsCg.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
if __name__ == '__main__':
55

6-
ifalscg = pyreclab.IFAlsConjugateGradient( dataset = 'dataset/lastfm-dataset.txt',
6+
ifalscg = pyreclab.IFAlsConjugateGradient( dataset = 'dataset/u1.base',
77
dlmchar = b'\t',
88
header = False,
99
usercol = 0,
@@ -16,8 +16,8 @@
1616
end = time.clock()
1717
print( 'training time: ' + str( end - start ) )
1818

19-
ranking = ifalscg.recommend( 'user_000436', 5, includeRated = False )
20-
print( 'recommendation for user user_000436: ' + str( ranking ) )
19+
ranking = ifalscg.recommend( '457', 5, includeRated = False )
20+
print( 'recommendation for user 457: ' + str( ranking ) )
2121

2222
print( '-> recommendation test' )
2323
start = time.clock()

Diff for: pypackage/__init__.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,12 @@
1-
from pyreclab.libpyreclab import MostPopular, UserAvg, ItemAvg, SlopeOne, UserKnn, ItemKnn, SVD
1+
from pyreclab.libpyreclab import MostPopular, UserAvg, ItemAvg, SlopeOne, UserKnn, ItemKnn, SVD, IFAls, IFAlsConjugateGradient
22

33
__all__ = [ 'MostPopular',
44
'UserAvg',
55
'ItemAvg',
66
'SlopeOne',
77
'UserKnn',
88
'ItemKnn',
9-
'SVD' ]
9+
'SVD',
10+
'IFAls',
11+
'IFAlsConjugateGradient' ]
1012

0 commit comments

Comments
 (0)