Update for HDBSCAN bug and train_valid_split error

environment.yml

@@ -23,4 +23,5 @@ dependencies:
     - pip
     - pip:
         - pyrolite
-        - PyIRoGlass
+        - PyIRoGlass
+        - hdbscan

mineralML_colab.ipynb

@@ -15,8 +15,8 @@
     "import pandas as pd\n",
     "from sklearn.metrics import classification_report\n",
     "\n",
-    "!pip install torch torchaudio torchvision torchtext\n",
-    "!pip install -i https://test.pypi.org/simple/ mineralML==0.0.0.0\n",
+    "!pip install Cython hdbscan\n",
+    "!pip install -i https://test.pypi.org/simple/ mineralML==0.0.0.4\n",
     "import mineralML as mm\n",
     "\n",
     "from google.colab import files\n",

setup.py

@@ -25,15 +25,16 @@
 
     package_data={
         # Include all pickle files
-        "": ["*.npz", "*.joblib", "*.pt"],
+        "": ["*.npz", "*.pt"],
     },
     install_requires=[
             'pandas',
             'numpy',
             'matplotlib',
             'scikit-learn',
             'scipy',
-            'torch'
+            'torch',
+            'hdbscan'
             ],
 
     classifiers=[

src/mineralML/unsupervised.py

@@ -9,7 +9,7 @@
 import pandas as pd
 
 from sklearn.preprocessing import StandardScaler
-from sklearn.model_selection import train_valid_split
+from sklearn.model_selection import train_test_split
 from sklearn.metrics import classification_report, confusion_matrix
 from imblearn.over_sampling import RandomOverSampler