mirror of
https://github.com/jung-geun/PSO.git
synced 2025-12-20 04:50:45 +09:00
23-07-11
mnist one hot 인코딩 적용후 손실 함수 mse 로 변경
This commit is contained in:
16
readme.md
16
readme.md
@@ -148,17 +148,17 @@ best_score = pso_iris.fit(
|
||||
3. mnist 문제
|
||||
|
||||
```python
|
||||
loss = 'sparse_categorical_crossentropy'
|
||||
loss = 'mse'
|
||||
|
||||
pso_mnist = Optimizer(
|
||||
model,
|
||||
loss=loss,
|
||||
n_particles=75,
|
||||
c0=0.35,
|
||||
c1=0.7,
|
||||
w_min=0.45,
|
||||
w_max=0.9,
|
||||
negative_swarm=0.2,
|
||||
c0=0.25,
|
||||
c1=0.4,
|
||||
w_min=0.2,
|
||||
w_max=0.6,
|
||||
negative_swarm=0.1,
|
||||
mutation_swarm=0.2,
|
||||
)
|
||||
|
||||
@@ -175,8 +175,8 @@ best_score = pso_mnist.fit(
|
||||
)
|
||||
```
|
||||
|
||||
위의 파라미터 기준 현재 정확도 46%를 보이고 있습니다
|
||||

|
||||
위의 파라미터 기준 현재 정확도 43.38%를 보이고 있습니다
|
||||

|
||||
|
||||
### Trouble Shooting
|
||||
|
||||
|
||||
Reference in New Issue
Block a user