xx
This commit is contained in:
parent
9cf3179317
commit
95b383a1b4
14
run.py
14
run.py
@ -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),
|
||||
]
|
||||
)
|
||||
|
1432
test-A/out.tsv
1432
test-A/out.tsv
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue
Block a user