Compare commits

..

1 Commits

Author SHA1 Message Date
5956cd0cd2 Update google colab setup in notebooks
Undate save_all in grapher to make dest. folder if it doesn't exist
2020-05-24 20:13:27 +02:00
6 changed files with 8 additions and 19 deletions

View File

@@ -9,10 +9,7 @@
"#\n", "#\n",
"# Google Colab setup\n", "# Google Colab setup\n",
"#\n", "#\n",
"#import os\n", "#!pip install git+https://github.com/doctorado-ml/stree"
"#os.chdir(\"/content\")\n",
"#!git clone https://github.com/Doctorado-ML/STree.git\n",
"#os.chdir(\"/content/STree\")"
] ]
}, },
{ {

View File

@@ -1,6 +1,6 @@
import setuptools import setuptools
__version__ = "0.9rc1" __version__ = "0.9rc2"
__author__ = "Ricardo Montañana Gómez" __author__ = "Ricardo Montañana Gómez"
def readme(): def readme():
@@ -12,7 +12,7 @@ setuptools.setup(
name='STree', name='STree',
version=__version__, version=__version__,
license='MIT License', license='MIT License',
description='a python interface to oblique decision tree implementations', description='Oblique decision tree with svm nodes',
long_description=readme(), long_description=readme(),
long_description_content_type='text/markdown', long_description_content_type='text/markdown',
packages=setuptools.find_packages(), packages=setuptools.find_packages(),
@@ -34,7 +34,6 @@ setuptools.setup(
'matplotlib', 'matplotlib',
'ipympl' 'ipympl'
], ],
data_files=[('data', ['data/.gitignore'])],
test_suite="stree.tests", test_suite="stree.tests",
zip_safe=False zip_safe=False
) )

View File

@@ -164,6 +164,8 @@ class Stree_grapher(Stree):
:type save_folder: str, optional :type save_folder: str, optional
""" """
self._check_fitted() self._check_fitted()
if not os.path.isdir(save_folder):
os.mkdir(save_folder)
seq = 1 seq = 1
for node in self: for node in self:
node.save_hyperplane(save_folder=save_folder, node.save_hyperplane(save_folder=save_folder,

View File

@@ -9,10 +9,7 @@
"#\n", "#\n",
"# Google Colab setup\n", "# Google Colab setup\n",
"#\n", "#\n",
"#import os\n", "#!pip install git+https://github.com/doctorado-ml/stree"
"#os.chdir(\"/content\")\n",
"#!git clone https://github.com/Doctorado-ML/STree.git\n",
"#os.chdir(\"/content/STree\")"
] ]
}, },
{ {

View File

@@ -9,10 +9,7 @@
"#\n", "#\n",
"# Google Colab setup\n", "# Google Colab setup\n",
"#\n", "#\n",
"#import os\n", "#!pip install git+https://github.com/doctorado-ml/stree"
"#os.chdir(\"/content\")\n",
"#!git clone https://github.com/Doctorado-ML/STree.git\n",
"#os.chdir(\"/content/STree\")"
] ]
}, },
{ {

View File

@@ -9,10 +9,7 @@
"#\n", "#\n",
"# Google Colab setup\n", "# Google Colab setup\n",
"#\n", "#\n",
"#import os\n", "#!pip install git+https://github.com/doctorado-ml/stree"
"#os.chdir(\"/content\")\n",
"#!git clone https://github.com/Doctorado-ML/STree.git\n",
"#os.chdir(\"/content/STree\")"
] ]
}, },
{ {