def num_to_return_of_this_sim_cat(self, sim_cat_idx): # sim_cat_idx 是这个sim_cat在所有相似性递减的sim_cat里面的下标值 return self.total_rec_num strategy = Strategy() if __name__ == "__main__": import dim_items_index_builder from dim_items_index_builder import * dim_items_index_builder = DimItemsIndexBuilder() dim_items_index_builder.build_from_file(common.dim_items_file) dim_items_index = dim_items_index_builder.get_res() print 'dim_items_index size:%d' % (len(dim_items_index)) cat_to_item_rindex_builder = CatToItemRindexBuilder() cat_to_item_rindex_builder.build_from_file(common.dim_items_file) cat_to_item_rindex = cat_to_item_rindex_builder.get_res() import cat_sim_builder from cat_sim_builder import * cat_sim_builder = CatSimBuilder(dim_items_index_builder.get_res()) cat_sim_builder.build_from_file(common.dim_fashion_matchsets_file) cat_sim_rindex_builder = CatSimRindexBuilder() cat_sim = cat_sim_builder.get_res() cat_sim_rindex_builder.build_from_CatSimBuilder(cat_sim_builder)
print >> fd, "%d %d" % (sim_cat_id, weight) def get_res(self): return self.cat_sim_rindex if __name__ == "__main__": import common import dim_items_index_builder from dim_items_index_builder import * dim_items_index_builder = DimItemsIndexBuilder() dim_items_index_builder.build_from_file(common.dim_items_file) dim_items_index = dim_items_index_builder.get_res() cat_sim_builder = CatSimBuilder(dim_items_index) cat_sim_builder.build_from_file(common.dim_fashion_matchsets_file) cat_sim = cat_sim_builder.get_res() cat_sim_rindex_builder = CatSimRindexBuilder() cat_sim_rindex_builder.build_from_CatSimBuilder(cat_sim_builder) cat_sim_rindex = cat_sim_rindex_builder.get_res() cat_sim_rindex_builder.dump_to_file(common.cat_sim_rindex_dump_file) print cat_sim_rindex print len(cat_sim_rindex)