bs/Step_Data_Project_India
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Removed changes for shorter run

Browse Source
This commit is contained in:
bs
2025-09-04 13:45:34 +02:00
parent b1b63d416d
commit 4adb507d6a
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
-1
View File
@@ -135,7 +135,6 @@ def main_two():
file_path = os.path.join(dataset_path, data_filename) file_path = os.path.join(dataset_path, data_filename)
df = load_dataset(file_path) df = load_dataset(file_path)
df = remove_covid_data(df) df = remove_covid_data(df)
df = df.head(1000) # TODO: remove
tr,val,te = split_method(df, percentages=(80,10,10)) tr,val,te = split_method(df, percentages=(80,10,10))
tr = reduce_columns(tr, data_filename) tr = reduce_columns(tr, data_filename)
val = reduce_columns(val, data_filename) val = reduce_columns(val, data_filename)
pipeline.py
+2 -2
View File
@@ -11,7 +11,7 @@ from tensorflow.keras.callbacks import ReduceLROnPlateau, EarlyStopping
from keras_tuner import RandomSearch from keras_tuner import RandomSearch
from sklearn.metrics import accuracy_score from sklearn.metrics import accuracy_score
epochs = 2#50 # TODO: change epochs = 50
model_type_gru = 'GRU' model_type_gru = 'GRU'
model_type_lstm = 'LSTM' model_type_lstm = 'LSTM'
model_type_bilstm = 'BiLSTM' model_type_bilstm = 'BiLSTM'
@@ -114,7 +114,7 @@ def train_models(user_data, user_data_val, sequence_lengths, tuner_dir="./workin
tuner = RandomSearch( tuner = RandomSearch(
build_model, build_model,
objective='val_loss', objective='val_loss',
max_trials=2, #30, TODO: change max_trials=30,
executions_per_trial=2, executions_per_trial=2,
directory=tuner_dir, directory=tuner_dir,
project_name=f'lstm_seq_{sequence_length}' project_name=f'lstm_seq_{sequence_length}'