readme 파일 수정 - env 파일 및 설명 추가 , 참고 자료 수정
iris_tf.py 모델의 성능 교차 검증을 위해 추가
This commit is contained in:
jung-geun
2023-06-23 06:37:01 +00:00
parent 953cd44396
commit 2a28b7fa04
7 changed files with 98 additions and 57 deletions

View File

@@ -11,8 +11,8 @@ from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from tensorflow import keras
from tensorflow.keras.models import Sequential
from tensorflow.keras import layers
from tensorflow.keras.models import Sequential
from pso import Optimizer
@@ -40,7 +40,7 @@ def load_data():
model = make_model()
x_train, x_test, y_train, y_test = load_data()
loss = ['categorical_crossentropy', 'accuracy','mse']
loss = ['categorical_crossentropy']
pso_iris = Optimizer(
model,