This commit is contained in:
pangkaicheng
2024-03-12 12:13:46 +08:00
parent b2be4484a5
commit 26c328b1e4

View File

@@ -8,9 +8,9 @@ if __name__ == '__main__':
with open("./test_param/recommendation_test.json", "r") as f:
param = json.load(f)
fashion_dataset = FashionDataset(param["database"])
service = OutfitMaterTypeAware()
for item in param["query"]:
outfits = fashion_dataset.generate_outfit(item, param["topk"], param["max_outfits"])
service = OutfitMaterTypeAware()
scores = service.get_result(outfits)
print(scores)
# service.visualize(outfits, scores, param["topk"], best=True,