CoolFace
Apppublic

Anjana-S/DeepLearning

sourceHugging Faceupdated 3y agoView on Hugging Face
0likes
imdb_perceptron.py22 linesDownload Raw Back to root
1from tensorflow.keras.datasets import imdb2from Perceptron import  Perceptron3from tensorflow.keras.preprocessing.sequence import pad_sequences4from sklearn.metrics import accuracy_score5import pickle6 7top_words = 50008(X_train, y_train), (X_test,y_test) = imdb.load_data(num_words=top_words)9 10max_review_length = 50011X_train = pad_sequences(X_train, maxlen=max_review_length)12X_test = pad_sequences(X_test, maxlen=max_review_length)13 14percep = Perceptron(epochs=100)15 16percep.fit(X_train, y_train)17pred = percep.predict(X_test)18 19print(f"Accuracy : {accuracy_score(pred, y_test)}")20 21with open(r'C:\Users\shahi\Desktop\My Projects\DeepPredictorHub\PP.pkl','wb') as file:22    pickle.dump(percep, file)