pyerualjetwork 2.7.8__py3-none-any.whl → 4.1.9__py3-none-any.whl

pyerualjetwork/data_operations.py

@@ -0,0 +1,403 @@
+from tqdm import tqdm
+import numpy as np
+from colorama import Fore, Style
+import sys
+import math
+
+def encode_one_hot(y_train, y_test=None, summary=False):
+    """
+    Performs one-hot encoding on y_train and y_test data.
+
+    Args:
+        y_train (numpy.ndarray): Train label data.
+        y_test (numpy.ndarray): Test label data one-hot encoded. (optional).
+        summary (bool, optional): If True, prints the class-to-index mapping. Default: False
+    
+    Returns:
+        tuple: One-hot encoded y_train and (if given) y_test.
+    """
+    from .memory_operations import optimize_labels
+    
+    y_train = optimize_labels(y_train, one_hot_encoded=False, cuda=False)
+    y_test = optimize_labels(y_test, one_hot_encoded=False, cuda=False)
+
+    classes = np.unique(y_train)
+    class_count = len(classes)
+
+    class_to_index = {cls: idx for idx, cls in enumerate(classes)}
+
+    if summary:
+        print("Class-to-index mapping:")
+        for cls, idx in class_to_index.items():
+            print(f"  {idx}: {cls}")
+
+    y_train_encoded = np.zeros((y_train.shape[0], class_count), dtype=y_train.dtype)
+    for i, label in enumerate(y_train):
+        y_train_encoded[i, class_to_index[label]] = 1
+
+    if y_test is not None:
+        y_test_encoded = np.zeros((y_test.shape[0], class_count), dtype=y_test.dtype)
+        for i, label in enumerate(y_test):
+            y_test_encoded[i, class_to_index[label]] = 1
+        return y_train_encoded, y_test_encoded
+
+    return y_train_encoded
+
+
+def decode_one_hot(encoded_data):
+    """
+    Decodes one-hot encoded data to original categorical labels.
+
+    Args:
+        encoded_data (numpy.ndarray): One-hot encoded data with shape (n_samples, n_classes).
+
+    Returns:
+        numpy.ndarray: Decoded categorical labels with shape (n_samples,).
+    """
+
+    decoded_labels = np.argmax(encoded_data, axis=1)
+
+    return decoded_labels
+
+
+def split(X, y, test_size, random_state=42, dtype=np.float32):
+    """
+    Splits the given X (features) and y (labels) data into training and testing subsets.
+
+    Args:
+        X (numpy.ndarray): Features data.
+        y (numpy.ndarray): Labels data.
+        test_size (float or int): Proportion or number of samples for the test subset.
+        random_state (int or None): Seed for random state. Default: 42.
+        dtype (numpy.dtype): Data type for the arrays. np.float32 by default. Example: np.float64 or np.float16. [fp32 for balanced devices, fp64 for strong devices, fp16 for weak devices: not reccomended!]
+
+    Returns:
+        tuple: x_train, x_test, y_train, y_test as ordered training and testing data subsets.
+    """
+    from .memory_operations import transfer_to_cpu, optimize_labels
+    
+    X = transfer_to_cpu(X, dtype=dtype)
+    y = optimize_labels(y, one_hot_encoded=False, cuda=False)
+
+    num_samples = X.shape[0]
+
+    if isinstance(test_size, float):
+        test_size = int(test_size * num_samples)
+    elif isinstance(test_size, int):
+        if test_size > num_samples:
+            raise ValueError(
+                "test_size cannot be larger than the number of samples.")
+    else:
+        raise ValueError("test_size should be float or int.")
+
+    if random_state is not None:
+        np.random.seed(random_state)
+
+    indices = np.arange(num_samples)
+    np.random.shuffle(indices)
+
+    test_indices = indices[:test_size]
+    train_indices = indices[test_size:]
+
+    x_train, x_test = X[train_indices], X[test_indices]
+    y_train, y_test = y[train_indices], y[test_indices]
+
+    del X, y
+
+    return x_train, x_test, y_train, y_test
+
+
+def manuel_balancer(x_train, y_train, target_samples_per_class, dtype=np.float32):
+    """
+    Generates synthetic examples to balance classes to the specified number of examples per class.
+
+    Arguments:
+    x_train -- Input dataset (examples) - NumPy array format
+    y_train -- Class labels (one-hot encoded) - NumPy array format
+    target_samples_per_class -- Desired number of samples per class
+    dtype (numpy.dtype): Data type for the arrays. np.float32 by default. Example: np.float64 or np.float16. [fp32 for balanced devices, fp64 for strong devices, fp16 for weak devices: not reccomended!]
+
+    Returns:
+    x_balanced -- Balanced input dataset (NumPy array format)
+    y_balanced -- Balanced class labels (one-hot encoded, NumPy array format)
+    """
+    from .ui import loading_bars
+    from .memory_operations import transfer_to_cpu
+    
+    x_train = transfer_to_cpu(x_train, dtype=dtype)
+
+    bar_format = loading_bars()[0]
+    classes = np.arange(y_train.shape[1])
+    class_count = len(classes)
+    
+    x_balanced = []
+    y_balanced = []
+
+    for class_label in tqdm(range(class_count),leave=False, ascii="▱▰",
+            bar_format=bar_format,desc='Augmenting Data',ncols= 52):
+        class_indices = np.where(np.argmax(y_train, axis=1) == class_label)[0]
+        num_samples = len(class_indices)
+        
+        if num_samples > target_samples_per_class:
+      
+            selected_indices = np.random.choice(class_indices, target_samples_per_class, replace=False)
+            x_balanced.append(x_train[selected_indices])
+            y_balanced.append(y_train[selected_indices])
+            
+        else:
+            
+            x_balanced.append(x_train[class_indices])
+            y_balanced.append(y_train[class_indices])
+
+            if num_samples < target_samples_per_class:
+                
+                samples_to_add = target_samples_per_class - num_samples
+                additional_samples = np.zeros((samples_to_add, x_train.shape[1]), dtype=x_train.dtype)
+                additional_labels = np.zeros((samples_to_add, y_train.shape[1]), dtype=y_train.dtype)
+                
+                for i in range(samples_to_add):
+
+                    random_indices = np.random.choice(class_indices, 2, replace=False)
+                    sample1 = x_train[random_indices[0]]
+                    sample2 = x_train[random_indices[1]]
+
+                    
+                    synthetic_sample = sample1 + (sample2 - sample1) * np.random.rand()
+
+                    additional_samples[i] = synthetic_sample
+                    additional_labels[i] = y_train[class_indices[0]]
+                    
+                    
+                x_balanced.append(additional_samples)
+                y_balanced.append(additional_labels)
+    
+    x_balanced = np.vstack(x_balanced, dtype=x_train.dtype)
+    y_balanced = np.vstack(y_balanced, dtype=y_train.dtype)
+
+    del x_train, y_train
+
+    return x_balanced, y_balanced
+
+
+def auto_balancer(x_train, y_train, dtype=np.float32):
+
+    """
+   Function to balance the training data across different classes.
+
+   Arguments:
+   x_train (list): Input data for training.
+   y_train (list): Labels corresponding to the input data. one-hot encoded.
+   dtype (numpy.dtype): Data type for the arrays. np.float32 by default. Example: np.float64 or np.float16. [fp32 for balanced devices, fp64 for strong devices, fp16 for weak devices: not reccomended!]
+
+   Returns:
+   tuple: A tuple containing balanced input data and labels.
+    """
+    from .ui import loading_bars
+    from .memory_operations import transfer_to_cpu
+    
+    x_train = transfer_to_cpu(x_train, dtype=dtype)
+
+    bar_format = loading_bars()[0]
+    classes = np.arange(y_train.shape[1])
+    class_count = len(classes)
+    
+    try:
+        ClassIndices = {i: np.where(y_train[:, i] == 1)[
+            0] for i in range(class_count)}
+        classes = [len(ClassIndices[i]) for i in range(class_count)]
+
+        if len(set(classes)) == 1:
+            print(Fore.WHITE + "INFO: Data have already balanced. from: auto_balancer" + Style.RESET_ALL)
+            return x_train, y_train
+
+        MinCount = min(classes)
+
+        BalancedIndices = []
+        for i in tqdm(range(class_count),leave=False, ascii="▱▰",
+            bar_format= bar_format, desc='Balancing Data',ncols=70):
+            if len(ClassIndices[i]) > MinCount:
+                SelectedIndices = np.random.choice(
+                    ClassIndices[i], MinCount, replace=False)
+            else:
+                SelectedIndices = ClassIndices[i]
+            BalancedIndices.extend(SelectedIndices)
+
+        BalancedInputs = [x_train[idx] for idx in BalancedIndices]
+        BalancedLabels = [y_train[idx] for idx in BalancedIndices]
+
+        permutation = np.random.permutation(len(BalancedInputs))
+        BalancedInputs = np.array(BalancedInputs)[permutation]
+        BalancedLabels = np.array(BalancedLabels)[permutation]
+
+        print(Fore.GREEN + "Data Succesfully Balanced from: " + str(len(x_train)
+                                                                                 ) + " to: " + str(len(BalancedInputs)) + ". from: auto_balancer " + Style.RESET_ALL)
+    except:
+        print(Fore.RED + "ERROR: Inputs and labels must be same length check parameters")
+        sys.exit()
+
+    BalancedInputs = BalancedInputs.astype(dtype, copy=False)
+    BalancedLabels = BalancedLabels.astype(dtype=y_train.dtype, copy=False)
+
+    del x_train, y_train
+
+    return BalancedInputs, BalancedLabels
+
+
+def synthetic_augmentation(x, y, dtype=np.float32):
+    """
+    Generates synthetic examples to balance classes with fewer examples.
+
+    Arguments:
+    x -- Input dataset (examples) - array format
+    y -- Class labels (one-hot encoded) - array format
+    dtype (numpy.dtype): Data type for the arrays. np.float32 by default. Example: np.float64 or np.float16. [fp32 for balanced devices, fp64 for strong devices, fp16 for weak devices: not reccomended!]
+
+    Returns:
+    x_balanced -- Balanced input dataset (array format)
+    y_balanced -- Balanced class labels (one-hot encoded, array format)
+    """
+    from .ui import loading_bars
+    from .memory_operations import transfer_to_cpu
+    
+    x = transfer_to_cpu(x, dtype=dtype)
+
+    bar_format = loading_bars()[0]
+    classes = np.arange(y.shape[1])
+    class_count = len(classes)
+
+    class_distribution = {i: 0 for i in range(class_count)}
+    for label in y:
+        class_distribution[np.argmax(label)] += 1
+
+    max_class_count = max(class_distribution.values())
+
+    x_balanced = list(x)
+    y_balanced = list(y)
+
+
+    for class_label in tqdm(range(class_count), leave=False, ascii="▱▰",
+            bar_format=bar_format,desc='Augmenting Data',ncols= 52):
+        class_indices = [i for i, label in enumerate(
+            y) if np.argmax(label) == class_label]
+        num_samples = len(class_indices)
+
+        if num_samples < max_class_count:
+            while num_samples < max_class_count:
+
+                random_indices = np.random.choice(
+                    class_indices, 2, replace=False)
+                sample1 = x[random_indices[0]]
+                sample2 = x[random_indices[1]]
+
+                synthetic_sample = sample1 + \
+                    (np.array(sample2) - np.array(sample1)) * np.random.rand()
+
+                x_balanced.append(synthetic_sample.tolist())
+                y_balanced.append(y[class_indices[0]])
+
+                num_samples += 1
+
+    x_balanced = np.array(x_balanced).astype(dtype, copy=False)
+    y_balanced = np.array(y_balanced).astype(dtype=y.dtype, copy=False)
+    
+    del x, y
+
+    return x_balanced, y_balanced
+
+
+def standard_scaler(x_train=None, x_test=None, scaler_params=None, dtype=np.float32):
+    """
+    Standardizes training and test datasets. x_test may be None.
+
+    Args:
+        x_train: numpy.ndarray
+        
+        x_test: numpy.ndarray (optional)
+        
+        scaler_params (optional for using model)
+        
+        dtype (numpy.dtype): Data type for the arrays. np.float32 by default. Example: np.float64 or np.float16. [fp32 for balanced devices, fp64 for strong devices, fp16 for weak devices: not reccomended!]
+
+    Returns:
+        list:
+        Scaler parameters: mean and std
+        tuple
+        Standardized training and test datasets
+    """
+    if x_train is not None and scaler_params is None and x_test is not None:
+        x_train = x_train.astype(dtype, copy=False)
+        x_test = x_test.astype(dtype, copy=False)
+
+        mean = np.mean(x_train, axis=0)
+        std = np.std(x_train, axis=0)
+        
+        train_data_scaled = (x_train - mean) / std
+        test_data_scaled = (x_test - mean) / std
+
+        train_data_scaled = np.nan_to_num(train_data_scaled, nan=0)
+        test_data_scaled = np.nan_to_num(test_data_scaled, nan=0)
+
+        scaler_params = [mean, std]
+
+        return scaler_params, train_data_scaled, test_data_scaled
+    
+    if scaler_params is None and x_train is None and x_test is not None:
+        return x_test.astype(dtype, copy=False)  # sample data not scaled
+            
+    if scaler_params is not None:
+        x_test = x_test.astype(dtype, copy=False)
+        scaled_data = (x_test - scaler_params[0]) / scaler_params[1]
+        scaled_data = np.nan_to_num(scaled_data, nan=0)
+
+        return scaled_data  # sample data scaled
+    
+    
+def normalization(
+    Input,  # num: Input data to be normalized.
+dtype=np.float32):
+    """
+    Normalizes the input data using maximum absolute scaling.
+
+    Args:
+        Input (num): Input data to be normalized.
+        dtype (numpy.dtype): Data type for the arrays. np.float32 by default. Example: np.float64 or np.float16. [fp32 for balanced devices, fp64 for strong devices, fp16 for weak devices: not reccomended!]
+
+    Returns:
+        (num) Scaled input data after normalization.
+    """
+
+    MaxAbs = np.max(np.abs(Input.astype(dtype, copy=False)))
+    return (Input / MaxAbs)
+
+
+def find_closest_factors(a):
+
+    root = int(math.sqrt(a))
+    
+    for i in range(root, 0, -1):
+        if a % i == 0:
+            j = a // i
+            return i, j
+        
+
+def batcher(x_test, y_test, batch_size=1):
+
+    if batch_size == 1:
+        return x_test, y_test
+    
+    y_labels = np.argmax(y_test, axis=1)
+
+    sampled_x, sampled_y = [], []
+    
+    for class_label in np.unique(y_labels):
+
+        class_indices = np.where(y_labels == class_label)[0]
+        
+        num_samples = int(len(class_indices) * batch_size)
+        
+        sampled_indices = np.random.choice(class_indices, num_samples, replace=False)
+        
+        sampled_x.append(x_test[sampled_indices])
+        sampled_y.append(y_test[sampled_indices])
+
+    return np.concatenate(sampled_x), np.concatenate(sampled_y)