Skip to content

tensorflow

임포트

import tensorflow as tf
from tensorflow.keras.layers import *

모델 정의

model = tf.keras.Sequential([
    Dense(1, activation='sigmoid')
])

훈련 설정

model.compile(loss='binary_crossentropy', 
              metrics=['accuracy'])

훈련

model.fit(x_train.A, y_train, epochs=10)

평가

model.evaluate(x_test.A, y_test)

가중치

weight, bias = model.trainable_weights
word_weight = pd.DataFrame({
    '단어': cv.get_feature_names_out(),
    '가중치': weight.numpy().flat
})

예측

prob = model.predict(x_test.A)

다층신경망

model = tf.keras.Sequential([
    Dense(32, activation='relu'),
    Dense(1, activation='sigmoid')
])