[Python] 리스트의 요소로 클래스 넣기

노하람·2022년 3월 29일
0
post-custom-banner

아닛... 이렇게 사용할 일이 없어서 설마 되나 하고 시도해봤는데 되는 게 있더라구요.
거두절미하고 예제 코드부터 보여드리겠습니다.

import sys, os
import tensorflow as tf

optimizer_hp = "Adam"
optimizer_list = [tf.keras.optimizers.RMSprop, tf.keras.optimizers.SGD, tf.keras.optimizers.Adam]


for i in range(3):
   if optimizer_hp in str(optimizer_list[i]):
       a = optimizer_list[i]
a

친구에게 리스트 안에 클래스를 넣을 수 있다고 신기해서 말했더니, 원래 배열에 클래스 넣어서 쓰지 않나고 하더라구요..
저는 아직도 갈 길이 멀군요, 그렇게 쓸 일이 방금 처음 생겼거든요😭

하이퍼 파라미터 튜닝 변수로 옵티마이저를 추가하려고 했는데, argument를 str으로 "RMSprop"같이 받다보니, 해당 argument를 모델의 옵티마이저로 집어넣어주려 할 때 클래스로 들어가야 하는데 str 값이라 쓰지 못하겠더라구요.
그래서 클래스를 담은 리스트를 하나 만들고 해당 str을 포함하고 있으면 클래스를 변수에 할당하여 사용하였습니다!

나중에라도 잘 활용할 때가 있을 것 같네요 ㅎㅎㅎ

profile
MLOps, MLE 직무로 일하고 있습니다😍
post-custom-banner

0개의 댓글