Myosotis-Researches 0.0.19__tar.gz → 0.0.21__tar.gz

{myosotis_researches-0.0.19 → myosotis_researches-0.0.21/Myosotis_Researches.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Myosotis-Researches
-Version: 0.0.19
+Version: 0.0.21
 Summary: A repository for storing my progress of researches.
 Home-page: https://github.com/Zeyu-Xie/Myosotis-Researches
 Author: Zeyu Xie

{myosotis_researches-0.0.19 → myosotis_researches-0.0.21}/Myosotis_Researches.egg-info/SOURCES.txt

@@ -51,5 +51,6 @@ myosotis_researches/CcGAN/train_128_output_10/utils.py
 myosotis_researches/CcGAN/utils/__init__.py
 myosotis_researches/CcGAN/utils/concat_image_horizontal.py
 myosotis_researches/CcGAN/utils/concat_image_vertical.py
+myosotis_researches/CcGAN/utils/install_datasets.py
 myosotis_researches/CcGAN/utils/make_h5.py
 myosotis_researches/CcGAN/utils/print_hdf5_structure.py

{myosotis_researches-0.0.19/Myosotis_Researches.egg-info → myosotis_researches-0.0.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Myosotis-Researches
-Version: 0.0.19
+Version: 0.0.21
 Summary: A repository for storing my progress of researches.
 Home-page: https://github.com/Zeyu-Xie/Myosotis-Researches
 Author: Zeyu Xie

{myosotis_researches-0.0.19 → myosotis_researches-0.0.21}/myosotis_researches/CcGAN/train_128_output_10/train_ccgan.py

@@ -79,7 +79,7 @@ def train_ccgan(kernel_sigma, kappa, train_images, train_labels, netG, netD, net
     # printed images with labels between the 5-th quantile and 95-th quantile of training labels
     n_row=10; n_col = 1
     z_fixed = torch.randn(n_row*n_col, dim_gan, dtype=torch.float).cuda()
-    selected_labels = np.array([0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9])
+    selected_labels = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
 
     y_fixed = np.zeros(n_row*n_col)
     for i in range(n_row):

myosotis_researches-0.0.21/myosotis_researches/CcGAN/utils/install_datasets.py

@@ -0,0 +1,43 @@
+import gdown
+from importlib import resources
+import os
+import rarfile
+
+# Paths
+datasets_dir = resources.files("myosotis_researches").joinpath("CcGAN", "datasets")
+
+# File ID dictionary
+file_id_dict = {
+    "Ra": "1DJfpvXTX1x6Ldf4nZzqzlVsQSx9iXo9I",
+    "MNIST": "1ZmTSX_wl68nsQI0ImfhpIsPClW_QrK1T",
+}
+
+
+# Function
+def install_datasets(datasets_name):
+
+    # Path
+    zipped_dataset_path = os.path.join(datasets_dir, f"{datasets_name}_datasets.rar")
+
+    # File ID
+    file_id = file_id_dict[datasets_name]
+
+    # URL
+    url = f"https://drive.google.com/uc?id={file_id}"
+
+    # Download
+    gdown.download(url, zipped_dataset_path, quiet=False, use_cookies=False)
+
+    # Unzip
+    with rarfile.RarFile(zipped_dataset_path) as rf:
+        if rf.needs_password():
+            unzip_password = input("Password: ")
+            rf.extractall(path=datasets_dir, pwd=unzip_password)
+        else:
+            rf.extractall(path=datasets_dir)
+
+    # Delete zipped datasets
+    os.remove(zipped_dataset_path)
+
+
+__all__ = ["install_datasets"]

{myosotis_researches-0.0.19 → myosotis_researches-0.0.21}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="Myosotis-Researches",
-    version="0.0.19",
+    version="0.0.21",
     description="A repository for storing my progress of researches.",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",