feature_scaling.py

import pandas as pd 
from data_description import DataDescription
from sklearn.preprocessing import MinMaxScaler, StandardScaler
class FeatureScaling:
    bold_text_start = "\033[1m"
    bold_text_end = "\033[0;0m"
    tasks = [
        "\n1. Perform Normalization(MinMax Scaler)",
        "2. Perform Standardization(Standard Scaler)",
        "3. Show the Dataset"
    ]
    tasks_normalization = [
        "\n1. Normalize a specific Column",
        "2. Normalize the whole Dataset",
        "3. Show the Dataset"
    ]
    tasks_standardization = [
        "\n1. Standardize a specific Column",
        "2. Standardize the whole Dataset",
        "3. Show the Dataset"
    ]
    def __init__(self, data):
        self.data = data
    def normalization(self):
        while(1):
            print("\nTasks (Normalization)")
            for task in self.tasks_normalization:
                print(task)
            while(1):
                try:
                    choice = int(input(("\n\nWhat you want to do? [enter -1 to go back]  ")))
                except ValueError:
                    print("Integer Value required. Try again....")
                    continue
                break
            if choice == -1:
                break
            elif choice == 1:
                print(self.data.dtypes)
                columns = input("Enter all the column"+ self.bold_text_start + "(s)" + self.bold_text_end + "you want to normalize [enter -1 to go back]  ").lower()
                if columns == "-1":
                    break
                for column in columns.split(" "):
                    try:
                        minValue = self.data[column].min()
                        maxValue = self.data[column].max()
                        self.data[column] = (self.data[column] - minValue)/(maxValue - minValue)
                    except:
                        print("\nNot possible...")
                print("Done....")
            elif choice == 2:
                try:
                    self.data = pd.DataFrame(MinMaxScaler().fit_transform(self.data))
                    print("Done.....")
                except:
                    print("\nString Columns are present. So, " + self.bold_text_start + "NOT" + self.bold_text_end + " possible.\nYou can try the first option though.")
            elif choice==3:
                DataDescription.showDataset(self)
            else:
                print("\nWrong choice!! Try again...")
        return
    def standardization(self):
        while(1):
            print("\nTasks (Standardization)")
            for task in self.tasks_standardization:
                print(task)
            while(1):
                try:
                    choice = int(input(("\n\nWhat you want to do? [enter -1 to go back]  ")))
                except ValueError:
                    print("Integer Value required. Try again.....")
                    continue
                break
            if choice == -1:
                break
            elif choice == 1:
                print(self.data.dtypes)
                columns = input("Enter all the column"+ self.bold_text_start + "(s)" + self.bold_text_end + "you want to normalize [enter -1 to go back]  ").lower()
                if columns == "-1":
                    break
                for column in columns.split(" "):
                    try:
                        mean = self.data[column].mean()
                        standard_deviation = self.data[column].std()
                        self.data[column] = (self.data[column] - mean)/(standard_deviation)
                    except:
                        print("\nNot possible....")
                print("Done....")
            elif choice == 2:
                try:
                    self.data = pd.DataFrame(StandardScaler().fit_transform(self.data))
                    print("Done.......")
                except:
                    print("\nString Columns are present. So, " + self.bold_text_start + "NOT" + self.bold_text_end + " possible. \nYou can try the first option though.")
                break
            elif choice==3:
                DataDescription.showDataset(self)
            else:
                print("\nWrong choice!! Try again...")
        return
    def scaling(self):
        while(1):
            print("\nTasks (Feature Scaling)")
            for task in self.tasks:
                print(task)
            while(1):
                try:
                    choice = int(input(("\n\nWhat you want to do? [enter -1 to go back]  ")))
                except ValueError:
                    print("Integer Value required. Try again.....")
                    continue
                break
            if choice == -1:
                break
            elif choice == 1:
                self.normalization()
            elif choice == 2:
                self.standardization()
            elif choice==3:
                DataDescription.showDataset(self)
            else:
                print("\nWrong choice!! Try again...")
        return self.data