[tensorflow / pytorch] 모델 저장 /불러오기 코드

Tensorflow 모델을 파일에 저장하는 코드 cnn.save("my_cnn.h5") save함수는 신경망의 구조 정보와 가중치 정보를 저장한다. 대용량 데이터를 저장하는데 널리 쓰이는 HDF5파일 형식을 사용하기 때문에 확장자를 h5로 지정한다. 모델을 다시 불러다 쓰는 코드 #신경망 구조와 가중치를 저장하고 있는 파일을 읽어옴 cnn=tf.keras.models.load_model('my_cnn.h5') 이렇게 해도 되고, 아니면 해당 API를 선언하고 사용해도됨 from tensorflow.keras.models import load_model model= load_model("파일명") 텐서플로는 학습하고 있는 모델객체가 (변수명이) cnn이면 cnn.save() 로 모델을 저장하는데 파이토치는..

• format_list_bulleted 프로그래밍/파이썬 Python
• · 2022. 1. 22.
• textsms

[tensorflow] CNN을 이용한 필기 숫자 인식

초창기 모델인 LeNet-5를 텐서플로로 재현하여 데이터를 인식한다. 컨볼루션 신경망의 유연성을 확인하기 위해 표전적인 빌딩 블록을 벗어난 신경망도 실험한다. LeNet-5 재현 MNIST를 인식하는 컨볼루션 신경망이다. import numpy as np import tensorflow as tf from tensorflow.keras.datasets import mnist from tensorflow.keras.models import Sequential from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense from tensorflow.keras.optimizers import Adam #데이터를 신경망에 입력할 형태로 변환 ..

• format_list_bulleted 프로그래밍/파이썬 Python
• · 2022. 1. 22.
• textsms

Pandas 라이브러리 관련 함수 정리

파이썬으로 머신러닝을 다루다 보면 이렇게 데이터프레임을 많이 다루는데 매일 찾아보기도 귀찮고 해서 데이터 추가, 행 검색, 중복 검색, 상관관계... 등등 자주쓰는 함수들을 정리하였다. 위 ipynb파일을 열면 더 깔끔하게 확인할 수 있다. In [201]: import re, json import pandas as pd import numpy as np import os import matplotlib.pyplot as plt import seaborn as sns %matplotlib inline In [202]: PATH = 'data/' In [203]: AGE_GENDER = 'age_gender_bkts.csv' COUNTRIY = 'countries.csv..

• format_list_bulleted 프로그래밍/파이썬 Python
• · 2022. 1. 11.
• textsms

[python] os 관련 함수

os.path.join 경로를 합칠 수 있음 import os #데이터 읽기 directory ="./imdb dataset" x=[]; y=[] for c in ['neg', 'pos']: curr =os.path.join(directory,c) print(curr) os.listdir("경로") 경로내의 모든 파일과 디렉토리 리스트를 반환 for filename in os.listdir("/content/test_images"): if 'jpg' not in filename: continue os.chdir 해당 경로로 이동 os.chdir('/content/test_images') os.isfile 파일이 존재하는가? os.remove 해당 파일 지우기 if os.path.isfile('word.m..

• format_list_bulleted 프로그래밍/파이썬 Python
• · 2022. 1. 11.
• textsms