Commit 72ac610ba7c10df4cd1d2b132c653c0e5f6c5862
1 parent
690cbed8
Exists in
master
and in
1 other branch
Added config option k_neighbors.
Showing
1 changed file
with
18 additions
and
6 deletions
Show diff stats
src/config.py
... | ... | @@ -40,7 +40,7 @@ class Config(): |
40 | 40 | self.output = "/dev/null" |
41 | 41 | self.survey_mode = 1 |
42 | 42 | self.axi = "/var/lib/apt-xapian-index/index" |
43 | -# self.dde_url = "http://dde.debian.net/dde/q/udd/packs/all/%s?t=json" | |
43 | + #self.dde_url = "http://dde.debian.net/dde/q/udd/packs/all/%s?t=json" | |
44 | 44 | self.dde_url = "http://46.4.235.200:8000/q/udd/packages/all/%s?t=json" |
45 | 45 | self.popcon_index = os.path.expanduser("~/.app-recommender/popcon_index") |
46 | 46 | self.popcon_dir = os.path.expanduser("~/.app-recommender/popcon_dir") |
... | ... | @@ -51,6 +51,9 @@ class Config(): |
51 | 51 | self.strategy = "cb" |
52 | 52 | self.weight = "bm25" |
53 | 53 | self.profile_size = 50 |
54 | + # options: maximal, voted, desktop | |
55 | + self.profiling = "maximal" | |
56 | + self.k_neighbors = 100 | |
54 | 57 | self.load_options() |
55 | 58 | self.set_logger() |
56 | 59 | |
... | ... | @@ -77,7 +80,9 @@ class Config(): |
77 | 80 | print " [ recommender ]" |
78 | 81 | print " -w, --weight=OPTION Search weighting scheme" |
79 | 82 | print " -s, --strategy=OPTION Recommendation strategy" |
80 | - print " -z, --profile_size=SIZE Size of user profile" | |
83 | + print " -z, --profilesize=k Size of user profile" | |
84 | + print " -f, --profiling=OPTION Profile filter strategy" | |
85 | + print " -n, --neighbors=k Size of neighborhood for collaboration" | |
81 | 86 | print "" |
82 | 87 | print " [ weight options ] " |
83 | 88 | print " trad = traditional probabilistic weighting" |
... | ... | @@ -132,12 +137,15 @@ class Config(): |
132 | 137 | self.strategy = self.read_option('recommender', 'strategy') |
133 | 138 | self.profile_size = int(self.read_option('recommender', |
134 | 139 | 'profile_size')) |
140 | + self.profiling = self.read_option('recommender', 'profiling') | |
141 | + self.k_neighbors = int(self.read_option('recommender', | |
142 | + 'k_neighbors')) | |
135 | 143 | |
136 | - short_options = "hdvo:a:e:p:m:ul:c:x:w:s:z:" | |
144 | + short_options = "hdvo:a:e:p:m:ul:c:x:w:s:z:f:n:" | |
137 | 145 | long_options = ["help", "debug", "verbose", "output=", |
138 | 146 | "axi=", "dde=", "popconindex=", "popcondir=", "indexmode=", |
139 | - "clustersdir=", "kmedoids=", "max_popcon=", "weight=", "strategy=", | |
140 | - "profile_size="] | |
147 | + "clustersdir=", "kmedoids=", "maxpopcon=", "weight=", "strategy=", | |
148 | + "profile_size=", "profiling=", "neighbors="] | |
141 | 149 | try: |
142 | 150 | opts, args = getopt.getopt(sys.argv[1:], short_options, |
143 | 151 | long_options) |
... | ... | @@ -178,7 +186,11 @@ class Config(): |
178 | 186 | elif o in ("-s", "--strategy"): |
179 | 187 | self.strategy = p |
180 | 188 | elif o in ("-z", "--profile_size"): |
181 | - self.strategy = int(p) | |
189 | + self.profile_size = int(p) | |
190 | + elif o in ("-z", "--profiling"): | |
191 | + self.profiling = p | |
192 | + elif o in ("-n", "--neighbors"): | |
193 | + self.k_neighbors = int(p) | |
182 | 194 | else: |
183 | 195 | assert False, "unhandled option" |
184 | 196 | ... | ... |