mirror of
https://github.com/jung-geun/PSO.git
synced 2025-12-20 04:50:45 +09:00
23-06-23
readme 파일 수정 - env 파일 및 설명 추가 , 참고 자료 수정 iris_tf.py 모델의 성능 교차 검증을 위해 추가
This commit is contained in:
4
iris.py
4
iris.py
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user