Skip to content

Commit 75b9066

Browse files
committed
refactor: Move ML model class from lstm_model.py to ml_model.py
1 parent f2a05c0 commit 75b9066

File tree

2 files changed

+1
-1
lines changed

2 files changed

+1
-1
lines changed

experiment/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from experiment.plot import save_plot
2424
from experiment.src.data_loader import read_detected_data, read_metadata, join_label, get_y_labels
2525
from experiment.src.features import prepare_data
26-
from external.fork.CredSweeper.experiment.src.lstm_model import MlModel
26+
from experiment.src.ml_model import MlModel
2727
from experiment.src.model_config_preprocess import model_config_preprocess
2828
from experiment.src.prepare_data import prepare_train_data, data_checksum
2929

0 commit comments

Comments
 (0)