Method for Hyperparameter Tuning of Image Classification with PyCaret

被引:0
|
作者
Arai, Kohei [1 ,2 ]
Shimazoe, Jin [2 ]
Oda, Mariko [2 ]
机构
[1] Saga Univ, Informat Sci Dept, Saga, Japan
[2] Kurume Inst Technol, Appl AI Res Lab, Kurume City, Fukuoka, Japan
关键词
PyCaret; extra trees classifier; AUC; gini; entropy; feature split; ROC curve; SPATIAL-RESOLUTION; MAXIMUM CURVATURE; EIGEN SPACE;
D O I
10.14569/IJACSA.2023.0140930
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
method for hyperparameter tuning of image classification with PyCaret is proposed. The application example compares 14 classification methods and confirms that Extra Trees Classifier has the best performance among them, AUC=0.978, Recall=0.879, Precision=0.969, F1=0.912, Time=0.609 bottom. The Extra Trees Classifier produces a large number of decision trees, similar to the random forest algorithm, but with random sampling of each tree and no permutation. This creates a dataset for each tree containing unique samples, and from the ensemble set of features a certain number of features are also randomly selected for each tree. The most important and unique property of the Extra Trees Classifier is that the feature split values are chosen randomly. Instead of using Gini or entropy to split the data to compute locally optimal values, the algorithm randomly selects split values. This makes the tree diverse and uncorrelated. i.e. the diversity of each tree. Therefore, it is considered that the classification performance is better than other classification methods. Parameter tuning of Extra Trees Classifier was performed, and training performance, test performance, ROC curve, accuracy rate characteristics, etc. were evaluated.
引用
收藏
页码:276 / 282
页数:7
相关论文
共 50 条
  • [1] Automated hyperparameter tuning for crack image classification with deep learning
    Ottoni, Andre Luiz Carvalho
    Souza, Artur Moura
    Novo, Marcela Silva
    [J]. SOFT COMPUTING, 2023, 27 (23) : 18383 - 18402
  • [2] Automated hyperparameter tuning for crack image classification with deep learning
    André Luiz Carvalho Ottoni
    Artur Moura Souza
    Marcela Silva Novo
    [J]. Soft Computing, 2023, 27 : 18383 - 18402
  • [3] Hyperparameter Tuning Deep Learning for Diabetic Retinopathy Fundus Image Classification
    Shankar, K.
    Zhang, Yizhuo
    Liu, Yiwei
    Wu, Ling
    Chen, Chi-Hua
    [J]. IEEE ACCESS, 2020, 8 : 118164 - 118173
  • [4] Hyperparameter tuning of convolutional neural networks for building construction image classification
    André Luiz Carvalho Ottoni
    Marcela Silva Novo
    Dayana Bastos Costa
    [J]. The Visual Computer, 2023, 39 : 847 - 861
  • [5] Hyperparameter tuning of convolutional neural networks for building construction image classification
    Carvalho Ottoni, Andre Luiz
    Novo, Marcela Silva
    Costa, Dayana Bastos
    [J]. VISUAL COMPUTER, 2023, 39 (03): : 847 - 861
  • [6] A novel method for diabetes classification and prediction with Pycaret
    Pawan Whig
    Ketan Gupta
    Nasmin Jiwani
    Hruthika Jupalle
    Shama Kouser
    Naved Alam
    [J]. Microsystem Technologies, 2023, 29 : 1479 - 1487
  • [7] A novel method for diabetes classification and prediction with Pycaret
    Whig, Pawan
    Gupta, Ketan
    Jiwani, Nasmin
    Jupalle, Hruthika
    Kouser, Shama
    Alam, Naved
    [J]. MICROSYSTEM TECHNOLOGIES-MICRO-AND NANOSYSTEMS-INFORMATION STORAGE AND PROCESSING SYSTEMS, 2023, 29 (10): : 1479 - 1487
  • [8] ACCELERATING HYPERPARAMETER TUNING OF A DEEP LEARNING MODEL FOR REMOTE SENSING IMAGE CLASSIFICATION
    Aach, Marcel
    Sedona, Rocco
    Lintermann, Andreas
    Cavallaro, Gabriele
    Neukirchen, Helmut
    Riedel, Morris
    [J]. 2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022), 2022, : 263 - 266
  • [9] Method for Hyperparameter Tuning of EfficientNetV2-based Image Classification by Deliberately Modifying Optuna Tuned Result
    Shimazoe, Jin
    Arai, Kohei
    Oda, Mari
    [J]. INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2023, 14 (12) : 463 - 468
  • [10] Optimizing Image Classification: Automated Deep Learning Architecture Crafting with Network and Learning Hyperparameter Tuning
    Ang, Koon Meng
    Lim, Wei Hong
    Tiang, Sew Sun
    Sharma, Abhishek
    Eid, Marwa M.
    Tawfeek, Sayed M.
    Khafaga, Doaa Sami
    Alharbi, Amal H.
    Abdelhamid, Abdelaziz A.
    [J]. BIOMIMETICS, 2023, 8 (07)