This commit is contained in:
ZarebaMichal 2022-05-21 19:52:04 +02:00
parent 9cf3179317
commit 95b383a1b4
2 changed files with 723 additions and 723 deletions

14
run.py
View File

@ -17,8 +17,9 @@ test_A = pd.read_csv("test-A/in.tsv", header=None, sep="\t")
def preprocess_data(df_to_process, main_df=None):
final_df = pd.get_dummies(df_to_process, columns=[1, 2, 3, 4])
final_df = pd.get_dummies(df_to_process, columns=[1, 3, 4])
final_df.drop(columns=[0], inplace=True)
final_df.drop(columns=[2], inplace=True)
# numeric = [3]
# sc = StandardScaler()
# final_df[numeric] = sc.fit_transform(final_df[numeric])
@ -53,13 +54,12 @@ y = y_expected[0]
model = tensorflow.keras.models.Sequential(
[
tensorflow.keras.layers.Dense(
128, activation="relu", input_dim=76, kernel_regularizer="l2"
128, activation="relu", input_dim=73,
),
tensorflow.keras.layers.Dropout(0.5),
tensorflow.keras.layers.Dense(128 // 2, activation="relu", kernel_regularizer="l2"),
tensorflow.keras.layers.Dense(128 // 4, activation="relu", kernel_regularizer="l2"),
tensorflow.keras.layers.Dense(128 // 8, activation="relu", kernel_regularizer="l2"),
tensorflow.keras.layers.Dense(32, activation="relu", kernel_regularizer="l2"),
tensorflow.keras.layers.Dense(128 // 2, activation="relu"),
tensorflow.keras.layers.Dense(128 // 4, activation="relu"),
tensorflow.keras.layers.Dense(128 // 8, activation="relu"),
tensorflow.keras.layers.Dense(32, activation="relu"),
tensorflow.keras.layers.Dense(1),
]
)

File diff suppressed because it is too large Load Diff