dataset_loader.py

import pandas as pd

# Function to load dataset (assuming it is a CSV file)
def load_dataset():
    try:
        df = pd.read_csv("datasets/Canstralian/ShellCommands.csv")  # Adjust to correct path
        return df
    except FileNotFoundError:
        print("Dataset file not found.")
        return None