def selfTest(): # Read and parse JSON data recipes = getClassifiedRecipes(TRAINING_FILE) cuisinier = Cuisinier() cuisinier.addRecipes(recipes) success = 0 for recipe in recipes: result = cuisinier.classifyRecipe(Recipe(recipe.id, recipe.ingredients)) if result.cuisine == recipe.cuisine: success += 1 print(str(result.id) + ":\t" + result.cuisine + " / " + recipe.cuisine) print("Self-test: " + str(success) + "/" + str(len(recipes)))
def test(): recipesToClassify = getRecipes(TEST_FILE) recipes = getClassifiedRecipes(TRAINING_FILE) cuisinier = Cuisinier() cuisinier.addRecipes(recipes) with open('submissionData.csv', 'wb') as fileToWrite: csv_writer = csv.writer(fileToWrite) csv_writer = csv.writer(fileToWrite, delimiter=',', quotechar='|', quoting=csv.QUOTE_MINIMAL) csv_writer.writerow(["id","cuisine"]) success = 0 for recipe in recipesToClassify: result = cuisinier.classifyRecipe(Recipe(recipe.id, recipe.ingredients)) csv_writer.writerow([recipe.id, result.cuisine])