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')
])