test poly keras

This commit is contained in:
ZarebaMichal 2022-05-22 13:09:54 +02:00
parent 6f3eeb63c4
commit ec296f7853
2 changed files with 726 additions and 726 deletions

12
run2.py
View File

@ -24,15 +24,15 @@ df = poly.fit_transform(x)
model = Sequential(
[
Dense(512, activation="relu", input_dim=2702, kernel_regularizer="l2"),
Dense(512, activation="relu", input_dim=2702),
tensorflow.keras.layers.BatchNormalization(),
Dense(512 // 2, activation="relu", kernel_regularizer="l2"),
Dense(512 // 2, activation="relu"),
tensorflow.keras.layers.BatchNormalization(),
Dense(512 // 4, activation="relu", kernel_regularizer="l2"),
Dense(512 // 4, activation="relu"),
tensorflow.keras.layers.BatchNormalization(),
Dense(512 // 8, activation="relu", kernel_regularizer="l2"),
Dense(512 // 8, activation="relu"),
tensorflow.keras.layers.BatchNormalization(),
Dense(32, activation="relu", kernel_regularizer="l2"),
Dense(32, activation="relu"),
tensorflow.keras.layers.BatchNormalization(),
Dense(1),
]
@ -41,7 +41,7 @@ model = Sequential(
model.compile(
loss="mean_squared_error", optimizer="adam", metrics=["mean_squared_error"]
)
model.fit(df, y, epochs=100)
model.fit(df, y, epochs=60)
x_test = pd.read_csv("test-A/in.tsv", sep="\t", names=in_columns)
df_train = pd.read_csv("train/in.tsv", names=in_columns, sep="\t")

File diff suppressed because it is too large Load Diff