def load_situations(self): from situation import generate reader = csv.reader(open(self.label_filepath)) situations = [] for row in reader: temperature = row[1] occasion = row[2] situation = generate(temperature, occasion) situations.append(situation) return np.array(situations)
import os import csv import viewer from path import root_user_data from dataset import UserDataManager from system import Recommender from situation import generate def print_in_csv(user, files, distances): writer = csv.writer(sys.stdout) for f, distance in zip(files, distances): original_image_path = os.path.join(user, f) writer.writerow([original_image_path, distance]) # plot results using matplotlib user, occasion, temperature = sys.argv[1:4] situation = generate(temperature, occasion) recommender = Recommender(user) distances, indices = recommender.predict(situation) manager = UserDataManager(user) basenames = manager.basenames() basenames = [basenames[i] for i in indices] print_in_csv(user, basenames, distances)