Exemple #1
0
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)))
Exemple #2
0
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])