fork download
  1. from reader import read_testing, read_training
  2. from keras.utils import to_categorical
  3. import numpy as np
  4. import autokeras as ak
  5. from sklearn.metrics import classification_report, confusion_matrix
  6. emotions_map = [4, 4, 0, 2, 4, 3, 1]
  7.  
  8. def reading_data():
  9. print('Reading data...')
  10. x_test, y_test = read_testing()
  11. x_train, y_train = read_training()
  12. print(y_train.shape)
  13.  
  14. emotions_count = len(set(emotions_map))
  15. # y_train = to_categorical(y_train, emotions_count)
  16. # y_test = to_categorical(y_test, emotions_count)
  17.  
  18. x_train = np.array(x_train, 'float32')
  19. y_train = np.array(y_train, 'float32')
  20. x_test = np.array(x_test, 'float32')
  21. y_test = np.array(y_test, 'float32')
  22.  
  23. x_train /= 255 #normalize inputs between [0, 1]
  24. x_test /= 255
  25.  
  26. x_train = x_train.reshape(x_train.shape[0], 48, 48, 1)
  27. x_train = x_train.astype('float32')
  28. x_test = x_test.reshape(x_test.shape[0], 48, 48, 1)
  29. x_test = x_test.astype('float32')
  30.  
  31. print(x_train.shape[0], 'train samples')
  32. print(x_test.shape[0], 'test samples')
  33.  
  34. return x_train, x_test, y_train, y_test
  35.  
  36. def create_model(x_train, x_test, y_train, y_test):
  37. model = ak.ImageClassifier(path="/automodels/", verbose=True)
  38.  
  39. #increasing time will increase the accuracy
  40. model.fit(x_train, y_train, time_limit=60*60*24)
  41. model.final_fit(x_train, y_train, x_test, y_test, retrain=True)
  42. score = model.evaluate(x_test, y_test)
  43. print("accuracy on test set is ",100*score)
  44. autokeras_predictions = autokeras_model.predict(x_test)
  45. confusion_matrix(y_test, autokeras_predictions)
  46.  
  47. #exporting model
  48. model.export_autokeras_model('best_auto_keras_model.h5')
  49. print("done exporting")
  50.  
  51. def main():
  52. x_train, x_test, y_train, y_test = reading_data()
  53. create_model(x_train, x_test, y_train, y_test)
  54.  
  55. if __name__ == '__main__': main()
Runtime error #stdin #stdout #stderr 0.02s 27712KB
stdin
Standard input is empty
stdout
Standard output is empty
stderr
Traceback (most recent call last):
  File "./prog.py", line 1, in <module>
    from reader import read_testing, read_training
ImportError: No module named 'reader'