Fix paths

gatewayd-io · Mar 7, 2024 · 0baa212 · 0baa212
1 parent 8b50331
commit 0baa212
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/training/train.py b/training/train.py
@@ -1,6 +1,7 @@
 # Deep Learning Model Training with LSTM
 
 import pandas as pd
+import tensorflow as tf
 from tensorflow.keras.preprocessing.text import Tokenizer
 from tensorflow.keras.preprocessing.sequence import pad_sequences
 from tensorflow.keras.models import Sequential
@@ -17,7 +18,7 @@
 import numpy as np
 
 # Load dataset
-data = pd.read_csv("../model/sqli_dataset.csv")
+data = pd.read_csv("../dataset/sqli_dataset.csv")
 
 # Define parameters
 MAX_WORDS = 10000
@@ -67,4 +68,4 @@
 print("ROC: {:.2f}%".format(tp / (tp + fn) * 100))
 
 # Save model as SavedModel format
-save_model(model, "../sqli_model")
+save_model(model, "../sqli_model/1")