junshan-kit 2.2.6__py2.py3-none-any.whl → 2.2.7__py2.py3-none-any.whl

junshan_kit/DataSets.py

@@ -12,7 +12,7 @@ import junshan_kit.kit
 from sklearn.preprocessing import StandardScaler
 
 
-def download_data():
+def download_data(data_name):
     from junshan_kit.kit import JianguoyunDownloaderFirefox, JianguoyunDownloaderChrome
 
     # User selects download method
@@ -26,20 +26,20 @@ def download_data():
         choice = input("Enter the number of your choice (1 or 2): ").strip()
 
         if choice == "1":
-            JianguoyunDownloaderFirefox(url=url).run()
+            JianguoyunDownloaderFirefox(url, f"./exp_data/{data_name}").run()
             print("✅ Download completed using Firefox")
             break
         elif choice == "2":
-            JianguoyunDownloaderChrome(url=url).run()
+            JianguoyunDownloaderChrome(url, f"./exp_data/{data_name}").run()
             print("✅ Download completed using Chrome")
             break
         else:
             print("❌ Invalid choice. Please enter 1 or 2.\n")
 
 
-def credit_card_fraud_detection(data_name = "Credit Card Fraud Detection"):
+def credit_card_fraud_detection(data_name = "Credit Card Fraud Detection", print_info = False):
 
-    csv_path = f'./exp_data/{data_name}' + 'creditcard.csv'
+    csv_path = f'./exp_data/{data_name}/creditcard.csv'
     drop_cols = []
     label_col = 'Class'
     label_map = {0: -1, 1: 1}
@@ -48,28 +48,15 @@ def credit_card_fraud_detection(data_name = "Credit Card Fraud Detection"):
         print('\n' + '*'*60)
         print(f"Please download the data.")
         print(csv_path)
-        download_data()
+        download_data(data_name)
         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()
-    cleaner.preprocess_dataset(csv_path, drop_cols, label_col, label_map)
-
+    df = cleaner.preprocess_dataset(csv_path, drop_cols, label_col, label_map, print_info=print_info)
 
-    assert False
+    return df
+    
     
-    dataloader = junshan_kit.DataProcessor.CSV_TO_Pandas()
-    dataloader.preprocess_dataset()
-    # Step 0: Load the dataset
-    csv_path = "creditcard.csv"
-    df = pd.read_csv(csv_path)
-
 
 
 def wine_and_food_pairing_dataset():

{junshan_kit-2.2.6.dist-info → junshan_kit-2.2.7.dist-info}/METADATA

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

junshan_kit-2.2.7.dist-info/RECORD

@@ -0,0 +1,7 @@
+junshan_kit/DataProcessor.py,sha256=eryVmS5BFZj8wjDN2QWVHqkbFgFuWU0HXV9s6TGf9QM,4442
+junshan_kit/DataSets.py,sha256=QOpHAJkDLWBXP31cG8BStrXZITJEBiTCB1brmAy4TBQ,1985
+junshan_kit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+junshan_kit/kit.py,sha256=tB1TpW9hW1EweK1RQwHOdUo7uG1QU4vSeyR0fdaSydo,9569
+junshan_kit-2.2.7.dist-info/METADATA,sha256=LNjAHPrUu_tJs4HrYk2fyC-3GSb3UQj1c_4WIVYv7Wo,329
+junshan_kit-2.2.7.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
+junshan_kit-2.2.7.dist-info/RECORD,,

junshan_kit-2.2.6.dist-info/RECORD

@@ -1,7 +0,0 @@
-junshan_kit/DataProcessor.py,sha256=eryVmS5BFZj8wjDN2QWVHqkbFgFuWU0HXV9s6TGf9QM,4442
-junshan_kit/DataSets.py,sha256=jaKB5kh1pOR-o97hab2G2r_YfH69Bs4zR2CkiOfpyss,2085
-junshan_kit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-junshan_kit/kit.py,sha256=tB1TpW9hW1EweK1RQwHOdUo7uG1QU4vSeyR0fdaSydo,9569
-junshan_kit-2.2.6.dist-info/METADATA,sha256=CqdL1Yui6UbnUt5teWg0EpYP8ofCdviwWK8EkM_MuKw,329
-junshan_kit-2.2.6.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
-junshan_kit-2.2.6.dist-info/RECORD,,