본문 바로가기

IT64

[AI] Pytorch 개요 및 GPU 사용 여부 체크하기 PyTorch는 머신러닝 프레임워크 중 하나입니다. Tensorflow도 머신러닝 프레임워크 중 하나죠. PyTorch를 사용하면 GPU 연동을 통해 효율적으로 딥러닝 모델을 개발할 수 있습니다. colab에서도 사용 가능합니다. import torch 다음과 같이 import하여 사용할 수 있습니다. GPU 사용 여부 체크하기 a = [[1,2],[3,4],[5,6]] x = torch.tensor(a) print(x.is_cuda) x = x.cuda() print(x.is_cuda) x = x.cpu() print(x.is_cuda) x.cuda() 라는 명령어로 GPU로 옮기고, cpu()라는 명령으로 CPU로 옮길 수 있습니다. 2022. 12. 22.
[Python] 파일 입출력 파이썬에서 파일 객체를 만들 때는 open, 파일 객체를 닫을 때는 close 함수를 사용합니다. 파이썬에서는 여러 가지 파일 모드를 제공합니다. 먼저 파일에 데이터를 쓸 때는 write 함수를 사용하여 여러 줄에 걸쳐 파일에 데이터를 작성할 수 있습니다. f = open("ex.txt", 'w') for line in range(1,5): data = f"{line} 번째 줄" f.write(data) f.close() 파일의 모든 줄을 한꺼번에 읽을 때는 readlines() 함수를 사용할 수 있습니다. 각 줄을 읽기 때문에 줄바꿈 기호가 포함됩니다. 이를 제거하기 위해 strip을 사용합니다. f = open("ex.txt", 'w') lines = f.readlines() for line in r.. 2022. 12. 22.
[Python] Enumerate, Zip 내장 함수 range로 정수 집합을 순회하여 반복문을 작성하곤한다. 하지만 정수가 아닌, 리스트의 경우에는 직접 루프를 실행하기도 한다. 예를 들어, fruit = ['apple','banana','orange'] for a in fruit: print(a) 코드의 결과로는 과일이 하나씩 출력이 된다. 파이썬은 아이템의 인덱스와 값을 알고싶은 경우, enumerate 라는 기능을 제공한다. fruit_list = ['apple', 'banana', 'orange'] for index, fruit in enumerate(fruit_list): print(f'{index} : {fruit}') 반복문을 병렬로 처리할 때 사용하기 좋은 것은 zip 함수이다. zip은 제네레이터로 이터레이터 두 개 이상을 감싼.. 2022. 12. 22.
[AI] 간단한 RNN 모델로 노래 가사 예측하기 안녕하세요! 오늘은 노래 가사를 데이터로 사용하여, 특정 단어가 주어졌을 때 다음 단어로 어떤 게 나올지 예측해보는 프로그램을 실습하겠습니다! 먼저, 필요한 라이브러리를 불러옵니다. import numpy as np from tensorflow.keras.layers import Embedding, Flatten, Dense, LSTM, SimpleRNN from tensorflow.keras.models import Sequential from tensorflow.keras.preprocessing.text import one_hot from tensorflow.keras.preprocessing.sequence import pad_sequences from tensorflow.keras.preproc.. 2022. 12. 18.

티스토리툴바