junshan-kit 2.4.1__py2.py3-none-any.whl → 2.4.3__py2.py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of junshan-kit might be problematic. Click here for more details.

Files changed (5) hide show
  1. junshan_kit/DataSets.py +1 -1
  2. junshan_kit/ExperimentHub.py +10 -0
  3. {junshan_kit-2.4.1.dist-info → junshan_kit-2.4.3.dist-info}/METADATA +1 -1
  4. {junshan_kit-2.4.1.dist-info → junshan_kit-2.4.3.dist-info}/RECORD +5 -5
  5. {junshan_kit-2.4.1.dist-info → junshan_kit-2.4.3.dist-info}/WHEEL +0 -0
junshan_kit/DataSets.py CHANGED
@@ -40,7 +40,7 @@ def _download_data(data_name, data_type):
40
40
  print("❌ Invalid choice. Please enter 1 or 2.\n")
41
41
 
42
42
  # unzip file
43
- junshan_kit.kit.unzip_file(f'./exp_data/{data_type}/{data_name}/{data_name}.zip', f'./exp_data/{data_name}')
43
+ junshan_kit.kit.unzip_file(f'./exp_data/{data_type}/{data_name}/{data_name}.zip', f'./exp_data/{data_type}/{data_name}')
44
44
 
45
45
  def _export_csv(df, data_name):
46
46
  path = f'./exp_data/{data_name}/'
junshan_kit/ExperimentHub.py CHANGED
@@ -324,5 +324,15 @@ class Train_Steps:
324
324
  raise NotImplementedError(f"{optimizer_name} is not supported.")
325
325
 
326
326
  return optimizer
327
+
328
+ def train(self, train_loader, optimizer, optimizer_name, Paras):
329
+ for epoch in range(self.args.e):
330
+
331
+ for index, (X, Y) in enumerate(train_loader):
332
+ X, Y = X.to(Paras["device"]), Y.to(Paras["device"])
333
+
334
+ if optimizer_name in ["SGD", "ADAM"]:
335
+ print(1)
336
+
327
337
 
328
338
 
{junshan_kit-2.4.1.dist-info → junshan_kit-2.4.3.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: junshan_kit
3
- Version: 2.4.1
3
+ Version: 2.4.3
4
4
  Summary: This is an optimization tool.
5
5
  Author-email: Junshan Yin <junshanyin@163.com>
6
6
  Requires-Dist: kaggle==1.7.4.5
{junshan_kit-2.4.1.dist-info → junshan_kit-2.4.3.dist-info}/RECORD RENAMED
@@ -1,7 +1,7 @@
1
1
  junshan_kit/ComOptimizers.py,sha256=MUgFnm1DbbvNKv5-7nHJCLOfq4VjoNk1KLRR-ji5rOA,4637
2
2
  junshan_kit/DataProcessor.py,sha256=rp1zOTOoF98idwGM_QRzr8yC9M5fj1uyfVhdmV02kyQ,8962
3
- junshan_kit/DataSets.py,sha256=rRaCPtlR5WvH0E1CAaaWbVkfS5QU12ak31VbREq_prE,8354
4
- junshan_kit/ExperimentHub.py,sha256=OEiYRponPxvqL6SPEvtRLuKF-nA9tbDcOq-cboBevTE,11165
3
+ junshan_kit/DataSets.py,sha256=FJu0B90eiKdKKHQJr_aLNkOvwk4Vhk8mN73jYHZVEjc,8366
4
+ junshan_kit/ExperimentHub.py,sha256=MKduxa7U16zMoavgS-lVOCL2ypcMLpAaD8k7JitNqRU,11493
5
5
  junshan_kit/Models.py,sha256=GRTunJON1vLQz2IxgsoOKvjP-3zSJJLuB3CkJTAiImo,6884
6
6
  junshan_kit/Print_Info.py,sha256=vogYcXvoGcRGZV-7svi_mtiCZH6c8d-RhbZLFrLbKr8,3012
7
7
  junshan_kit/SPBM.py,sha256=UNzuyiEvMnhB5S7tzheh2jwpy-yIOngjagzqJOk3v8g,13723
@@ -11,6 +11,6 @@ junshan_kit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
11
11
  junshan_kit/check_args.py,sha256=7m4xSPAwqqQ0SPeKc-MCewDIDB7kFgsNYS2AuTTzGtk,3599
12
12
  junshan_kit/datahub.py,sha256=4c3P2TORMZ4va6NrSiojDCpnY_CGDlJV-5PG3u1_Isk,9081
13
13
  junshan_kit/kit.py,sha256=hpA4Zpn1VAuhdJSBBXswVum0CSk6QnB05GGLYoaRatQ,9792
14
- junshan_kit-2.4.1.dist-info/METADATA,sha256=4oSPIm8c_FRSDoZOjCS-XBrU1JRa0OKyYNtQe73senU,266
15
- junshan_kit-2.4.1.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
16
- junshan_kit-2.4.1.dist-info/RECORD,,
14
+ junshan_kit-2.4.3.dist-info/METADATA,sha256=YZLdpsItctrrUyByIXIS8pTtz_ZPuBgEaAbUyaE1-K8,266
15
+ junshan_kit-2.4.3.dist-info/WHEEL,sha256=tkmg4JIqwd9H8mL30xA7crRmoStyCtGp0VWshokd1Jc,105
16
+ junshan_kit-2.4.3.dist-info/RECORD,,