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)
