Aide pour un projet de perceptron

Bonjour,

Pour un travail, nous devons concevoir un perceptron (à deux classes) d'image provenant de google Quickdraw.
Notre professeur nous a fourni une fonction qui convertit ces images en vecteurs.
Voici la fonction :

def readDatasets(filename1, filename2):
    import numpy as np
    import random
    i = 0
    L = [[],[]]
    m = None
    for fn in [ filename1, filename2 ]:
        print('Lecture du fichier', fn)        
        L[i] = np.load(fn)
        if m == None:
            m = len(L[i])
        else:    
            m = min(m, len(L[i]))
        i += 1       
    print('Mélange aléatoire des images')      
    for i in range(2):
        np.random.shuffle(L[i])        
    s = m // 2    
    train = np.zeros((2*s, 785))                                                                 
    test = np.zeros((2*s, 785)) 
    for i in range(2):
        if i == 0:
            c = 1
        else:
            c = -1
        for j in range(s):
            train[i*s + j] = np.concatenate((np.array([c]), L[i][j]))     
            test[i*s + j] = np.concatenate((np.array([c]), L[i][s+j]))                                                                                                                               
    np.random.shuffle(train)
    np.random.shuffle(test)
    train_img = np.zeros((2*s, 784)) 
    test_img = np.zeros((2*s, 784)) 
    train_sol = np.zeros((2*s,)) 
    test_sol = np.zeros((2*s,)) 
    for j in range(2*s):
        train_img[j] = train[j][1:]
        test_img[j] = test[j][1:]
        train_sol[j] = train[j][0]
        test_sol[j] = test[j][0]
    return train_img, train_sol, test_img, test_sol    

Notre problème est que nous ne savons pas comment importer les fichiers npy fournis dans la fonction et que faire avec les résultats (train_img,train_sol,test_img,test_sol) obtenus.

Pour une meilleure explication, vous trouverez l'énoncé du projet en pièce jointe.


Merci d'avance pour votre aide,