thundersvm
We have recently released ThunderGBM, a fast GBDT and Random Forest library on GPUs.
add scikit-learn interface, see here
The mission of ThunderSVM is to help users easily and efficiently apply SVMs to solve problems. ThunderSVM exploits GPUs and multi-core CPUs to achieve high efficiency. Key features of ThunderSVM are as follows.
Support all functionalities of LibSVM such as one-class SVMs, SVC, SVR and probabilistic SVMs.
Use same command line options as LibSVM.
Supported Operating Systems: Linux, Windows and MacOS.
Why accelerate SVMs: A survey conducted by Kaggle in 2017 shows that 26% of the data mining and machine learning practitioners are users of SVMs.
Documentation | Installation | API Reference (doxygen)
cmake 2.8 or above
gcc 4.8 or above for Linux and MacOS
Visual C++ for Windows
If you want to use GPUs, you also need to install CUDA.
CUDA 7.5 or above
Download the Python wheel file (For Python3 or above).
For Linux
pip install thundersvm
for CUDA 9.0 - linux_x86_64
For Windows (64bit)
Install the Python wheel file.
pip install thundersvm-cu90-0.2.0-py3-none-linux_x86_64.whl
from thundersvm import SVCclf = SVC() clf.fit(x, y)
git clone https://github.com/Xtra-Computing/thundersvm.git
cd thundersvm mkdir build && cd build && cmake .. && make -j
If you run into issues that can be traced back to your version of gcc, use cmake
with a version flag to force gcc 6. That would look like this:
cmake -DCMAKE_C_COMPILER=gcc-6 -DCMAKE_CXX_COMPILER=g++-6 ..
# in thundersvm root directorygit submodule init eigen && git submodule update mkdir build && cd build && cmake -DUSE_CUDA=OFF -DUSE_EIGEN=ON .. && make -j
If make -j
doesn't work, please simply use make
. The number of CPU cores to use can be specified by the -o
option (e.g., -o 10
), and refer to Parameters for more information.
./bin/thundersvm-train -c 100 -g 0.5 ../dataset/test_dataset.txt ./bin/thundersvm-predict ../dataset/test_dataset.txt test_dataset.txt.model test_dataset.predict
You will see Accuracy = 0.98
after successful running.
If you use ThunderSVM in your paper, please cite our work (full version).
@article{wenthundersvm18, author = {Wen, Zeyi and Shi, Jiashuai and Li, Qinbin and He, Bingsheng and Chen, Jian}, title = {{ThunderSVM}: A Fast {SVM} Library on {GPUs} and {CPUs}}, journal = {Journal of Machine Learning Research}, volume={19}, pages={797--801}, year = {2018} }
Zeyi Wen, Jiashuai Shi, Bingsheng He, Yawen Chen, and Jian Chen. Efficient Multi-Class Probabilistic SVMs on GPUs. IEEE Transactions on Knowledge and Data Engineering (TKDE), 2018.
Zeyi Wen, Bingsheng He, Kotagiri Ramamohanarao, Shengliang Lu, and Jiashuai Shi. Efficient Gradient Boosted Decision Tree Training on GPUs. The 32nd IEEE International Parallel and Distributed Processing Symposium (IPDPS), pages 234-243, 2018.
We acknowledge NVIDIA for their hardware donations.
This project is hosted by NUS, collaborating with Prof. Jian Chen (South China University of Technology). Initial work of this project was done when Zeyi Wen worked at The University of Melbourne.
This work is partially supported by a MoE AcRF Tier 1 grant (T1 251RES1610) in Singapore.
We also thank the authors of LibSVM and OHD-SVM which inspire our algorithmic design.
Scene Graphs for Interpretable Video Anomaly Classification (published in NeurIPS18)
3D Semantic Segmentation for High-resolution Aerial Survey Derived Point Clouds using Deep Learning (published in SIGSPATIAL’18)
Accounting for part pose estimation uncertainties during trajectory generation for part pick-up using mobile manipulators. (published in International Conference on Robotics and Automation (ICRA), 2019).
还没有评论,说两句吧!
热门资源
Keras-ResNeXt
Keras ResNeXt Implementation of ResNeXt models...
seetafaceJNI
项目介绍 基于中科院seetaface2进行封装的JAVA...
spark-corenlp
This package wraps Stanford CoreNLP annotators ...
capsnet-with-caps...
CapsNet with capsule-wise convolution Project ...
inferno-boilerplate
This is a very basic boilerplate example for pe...
智能在线
400-630-6780
聆听.建议反馈
E-mail: support@tusaishared.com