반응형
- [ 인공지능/프레임워크 or 라이브러리 ]인공 신경망 코드로 구현해서 다중 분류해보기 (2)2024-03-17 12:02:55오늘 저녁 약속이 있는데 잘 하고 올게요~ 오늘은 지난번에 했던 데이터 세트와 다른 데이터로 복습할겸 인공신경망을 구성해서 다시 학습을 해볼 것이다 추가로 학습이 완료된 모델을 저장하는 방법도 실습해볼 것이다 모델을 저장하는 방법에는 두가지가 있다 모델의 파라미터만 저장하는 방법 이 방법은 모델의 계층 구조를 알고 있어야만 나중에 다시 해당 모델을 사용할 수 있음 모델 전체를 저장하는 방법 사용할 데이터 세트 : Fashion MNIST torchvision.datasets — Torchvision master documentation torchvision.datasets All datasets are subclasses of torch.utils.data.Dataset i.e, they have __g..
- [ 인공지능/프레임워크 or 라이브러리 ]데이터 로더, 모델 실습, 로깅2024-02-21 07:13:11가보자 가보자 1. 데이터 로더 (Data Loader) 데이터 세트가 항상 내가 원하는 형식으로 되어있지 않기 때문에 데이터를 불러올 때 특정한 형식으로 불러올 수 있어야 한다 이때 커스텀 데이터 로더를 구성해서 이용할 수 있다 다음 같은 구성으로 데이터 로더를 만들 수 있다 1-1. 꽃 분류 데이터 세트 불러오기 꽃 분류 데이터 세트 구성 커스텀 데이터 로더 구성하기 import torch from torchvision import transforms from torch.utils.data import DataLoader from PIL import Image from IPython.display import display import matplotlib.pyplot as plt import nump..
반응형