pyerualjetwork 4.3.8.dev15__py3-none-any.whl → 4.3.9b0__py3-none-any.whl

pyerualjetwork_afterburner/data_operations.py

@@ -1,406 +0,0 @@
-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,).
-    """
-
-    if encoded_data.ndim == 1: return np.argmax(encoded_data)
-    else: return np.argmax(encoded_data, axis=1)
-
-
-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)