import testing import generator rating_list = generator.create_lists(10, 30) permutation_list = generator.generate_row_permutation(10, 15) signature_matrix = testing.minhash(rating_list, permutation_list, 15)
import testing
import generator
# Various sizes of data set
num_people = 100
num_items = 100
num_permutations = 25
size_of_permutation = 20
user = 0 # who we want recommendations for
rating_list = generator.create_lists(num_people, num_items)
permutation_list = generator.generate_row_permutation(num_permutations, size_of_permutation, num_items)
signature_matrix = testing.minhash(rating_list, permutation_list, num_items)
# Output information to file
def write_results():
fd = open("output.txt", 'w')
fd.write("Ratings:\n")
for i in range(num_people):
fd.write("Person# %d, Ratings: %s\n" % (i, rating_list[i]))
fd.write("\nPermutation list:\n")
for i in range(num_permutations):
fd.write("Permutation# %d = %s\n" % (i, permutation_list[i]))
fd.write("\nSignature Matrix:\n")
for row in signature_matrix:
fd.write("%s\n" % row)
