junshan-kit 2.3.0__py2.py3-none-any.whl → 2.3.1__py2.py3-none-any.whl

junshan_kit/DataProcessor.py

@@ -18,6 +18,7 @@ class CSV_TO_Pandas:
         drop_cols: list,
         label_col: str,
         label_map: dict,
+        data_name: str,
         print_info=False,
     ):
         """
@@ -97,7 +98,7 @@ class CSV_TO_Pandas:
 
             # Step 6: Print dataset information
             print("\n" + "=" * 80)
-            print(f"{'Dataset Info':^70}")
+            print(f"{'{data_name} - Info':^70}")
             print("=" * 80)
             print(f"{'Original size:':<40} {m_original} rows x {n_original} cols")
             print(

junshan_kit/DataSets.py

@@ -45,7 +45,7 @@ def _run(csv_path, data_name, drop_cols, label_col, label_map, print_info):
         junshan_kit.kit.unzip_file(f'./exp_data/{data_name}/{data_name}.zip', f'./exp_data/{data_name}')
         
     cleaner = junshan_kit.DataProcessor.CSV_TO_Pandas()
-    df = cleaner.preprocess_dataset(csv_path, drop_cols, label_col, label_map, print_info=print_info)
+    df = cleaner.preprocess_dataset(csv_path, drop_cols, label_col, label_map, data_name, print_info=print_info)
 
     return df
 
@@ -67,7 +67,7 @@ def credit_card_fraud_detection(data_name = "Credit Card Fraud Detection", print
     return df
 
 
-def diabetes_health_indicators_dataset(data_name = "Diabetes Health Indicators Dataset", print_info = False):
+def diabetes_health_indicators_dataset(data_name = "Diabetes Health Indicators", print_info = False):
     csv_path = f'./exp_data/{data_name}/diabetes_dataset.csv'
     drop_cols = []
     label_col = 'diagnosed_diabetes'
@@ -78,7 +78,7 @@ def diabetes_health_indicators_dataset(data_name = "Diabetes Health Indicators D
     return df
 
 
-def electric_vehicle_population_data(data_name = "Electric Vehicle Population Data", print_info = False):
+def electric_vehicle_population_data(data_name = "Electric Vehicle Population", print_info = False):
     csv_path = f'./exp_data/{data_name}/Electric_Vehicle_Population_Data.csv'
     drop_cols = ['VIN (1-10)', 'DOL Vehicle ID', 'Vehicle Location']
     label_col = 'Electric Vehicle Type'
@@ -91,7 +91,7 @@ def electric_vehicle_population_data(data_name = "Electric Vehicle Population Da
 
     return df
 
-def global_house_purchase_dataset(data_name = "Global House Purchase Dataset", print_info = False):
+def global_house_purchase_dataset(data_name = "Global House Purchase", print_info = False):
     csv_path = f'./exp_data/{data_name}/global_house_purchase_dataset.csv'
     drop_cols = ['property_id']
     label_col = 'Electric Vehicle Type'
@@ -102,7 +102,7 @@ def global_house_purchase_dataset(data_name = "Global House Purchase Dataset", p
     return df
 
 
-def health_lifestyle_dataset(data_name = "Health_lifestyle_dataset", print_info = False):
+def health_lifestyle_dataset(data_name = "Health Lifestyle", print_info = False):
     csv_path = f'./exp_data/{data_name}/health_lifestyle_dataset.csv'
     drop_cols = ['id']
     label_col = 'decision'

{junshan_kit-2.3.0.dist-info → junshan_kit-2.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: junshan_kit
-Version: 2.3.0
+Version: 2.3.1
 Summary: This is an optimization tool.
 Author-email: Junshan Yin <junshanyin@163.com>
 Requires-Dist: kaggle==1.7.4.5

junshan_kit-2.3.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+junshan_kit/DataProcessor.py,sha256=QnYsqt2j4amZ4U04Urcu91RJs0du-tkl0N2lwxnTy1U,4472
+junshan_kit/DataSets.py,sha256=2P2AMQjQDKx8FITa5cNBaiFuUhPuWXKkgdlBM02LfPQ,3954
+junshan_kit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+junshan_kit/kit.py,sha256=tB1TpW9hW1EweK1RQwHOdUo7uG1QU4vSeyR0fdaSydo,9569
+junshan_kit-2.3.1.dist-info/METADATA,sha256=cCVc5fHfyWVno2RFPc8xskJ7XO8e7sWHYmV-8udM5s4,329
+junshan_kit-2.3.1.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
+junshan_kit-2.3.1.dist-info/RECORD,,

junshan_kit-2.3.0.dist-info/RECORD

@@ -1,7 +0,0 @@
-junshan_kit/DataProcessor.py,sha256=eryVmS5BFZj8wjDN2QWVHqkbFgFuWU0HXV9s6TGf9QM,4442
-junshan_kit/DataSets.py,sha256=rf5AVlA9DxP7wBpXjSO1_xznCMuxEoK50TqExafwHhc,3972
-junshan_kit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-junshan_kit/kit.py,sha256=tB1TpW9hW1EweK1RQwHOdUo7uG1QU4vSeyR0fdaSydo,9569
-junshan_kit-2.3.0.dist-info/METADATA,sha256=9NlU4YOD0zx5F5tLIMREKKxn-LwYD8-7IFVtp7DvMNM,329
-junshan_kit-2.3.0.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
-junshan_kit-2.3.0.dist-info/RECORD,,