hzhang/InverseOfLife.NeuralSolver
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add: data generator

Browse Source
This commit is contained in:
h z
2025-01-29 21:24:10 +00:00
parent 34e9b95145
commit 05d375d521
5 changed files with 101 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
NeuralSolver/NeuralSolver.py
View File

@@ -1,7 +1,7 @@
import numpy as np
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import regularizers
from Board import Board
import random
import os
@@ -29,8 +29,10 @@ class NeuralSolver:
def _build_reverse_model(self):
inputs = keras.Input(shape=(self.Width, self.Height, 1), name="FinalState")
hidden = keras.layers.Conv2D(32, 3, padding="same", activation="relu")(inputs)
hidden = keras.layers.Conv2D(32, 3, padding="same", activation="relu")(hidden)
hidden = keras.layers.Conv2D(32, 3, padding="same", activation="relu",kernel_regularizer=regularizers.l2(1e-4))(inputs)
hidden = keras.layers.BatchNormalization()(hidden)
hidden = keras.layers.Conv2D(32, 5, padding="same", activation="relu",kernel_regularizer=regularizers.l2(1e-4))(hidden)
hidden = keras.layers.Conv2D(32, 3, padding="same", activation="relu",kernel_regularizer=regularizers.l2(1e-4))(hidden)
outputs = keras.layers.Conv2D(1, 1, padding="same", activation="sigmoid")(hidden)
self.ReverseModel = keras.Model(inputs, outputs, name="ReverseModel")
@@ -102,7 +104,7 @@ class NeuralSolver:
def spec(self):
return f"{self.Width}x{self.Height}x{self.QuotientX}x{self.QuotientY}"
return f"{self.Width}x{self.Height}x{self.QuotientX}x{self.QuotientY}.weights"
def save_model(self):