Deep Learning
Simply Perceptron Implement for TF (CSV)
Rogue
2022. 9. 13. 23:46
반응형
import tensorflow as tf
import pandas as pd
import numpy as np
data = pd.read_csv('gpascore.csv')
data = data.dropna()
y = data['admit'].values
x = []
for i, rows in data.iterrows():
x.append([rows['gre'], rows['gpa'], rows['rank']])
model = tf.keras.models.Sequential([
tf.keras.layers.Dense(128, activation='sigmoid'),
tf.keras.layers.Dense(356, activation='sigmoid'),
tf.keras.layers.Dense(712, activation='sigmoid'),
tf.keras.layers.Dense(1, activation='sigmoid')
])
model.compile(optimizer='adam', loss='binary_crossentropy',
metrics=['accuracy'])
model.fit(np.array(x), np.array(y), epochs=1000)
# predict
predict_v = model.predict([[750, 3.70, 3], [400, 2.2, 1]])
print(predict_v)
반응형