와인 품종 예측


와인 품종 예측

비뉴 베르드(Vinho Verde) 지방에서 만들어진 레드와인 샘플 1,599개와 화이트 와인 4,898개의 등급, 맛, 산도를 측정해 만든 데이터를 이용해 품종을 예측하는 코드입니다. 학습한 모델을 저장하고 다시 불러오면서 비교된 수치를 바탕으로 오버피팅을 방지하고 불필요한 작업이 중단되는 과정이 담겨있습니다. 케라스와 판다스, 넘파이 등과 함께 이번 코드에서 쓰일 ModelCheckpoint와 EarlyStopping 모듈도 임포트 해줍니다. csv 데이터를 불러오는 과정에서 sample() 함수를 이용해서 전체 샘플의 15%로 학습을 하도록 설정합니다. 후술할 Epoch 횟수가 커지면서 학습 시간이 너무 길어지기 때문에 결과 출력의 속도 향상을 목적으로 해당 기능을 사용했습니다. 퍼셉트론은 총 4개의 층으로 이루어져 있으며, 렐루와 시그모이드 함수를 사용하여 학습합니다. 학습이 진행되면서 hdf5 확장자를 가진 모델이 Epoch마다 실행된 결과가 파일로 저장되는데 학습 정확도는...


#딥러닝 #머신러닝 #텐서플로우 #파이썬 #과적합 #와인품종예측 #인공지능 #테스트셋 #학습셋

원문링크 : 와인 품종 예측