Table 2.
Different models performance for the second dose vaccines’ side effect prediction.
| Training Set |
Validation Set |
Test Set |
|||||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Logistic Regression | SVM | XGBClassifier | RF | KNN | MLP | Logistic Regression | SVM | XGBClassifier | RF | KNN | MLP | Logistic Regression | SVM | XGBClassifier | RF | KNN | MLP | ||
| AZD1222 | Fever | 0.823 | 0.812 | 0.912 | 0.957 | 1.000 | 0.810 | 0.792 | 0.791 | 0.792 | 0.794 | 0.760 | 0.787 | 0.815 | 0.803 | 0.820 | 0.828 | 0.796 | 0.821 |
| Fatigue | 0.817 | 0.773 | 0.935 | 0.936 | 1.000 | 0.810 | 0.795 | 0.773 | 0.806 | 0.806 | 0.776 | 0.797 | 0.789 | 0.777 | 0.797 | 0.813 | 0.797 | 0.791 | |
| Headache | 0.855 | 0.832 | 0.904 | 0.954 | 1.000 | 0.856 | 0.837 | 0.821 | 0.827 | 0.838 | 0.814 | 0.840 | 0.839 | 0.803 | 0.832 | 0.850 | 0.825 | 0.842 | |
| Nausea | 0.885 | 0.860 | 0.918 | 0.929 | 1.000 | 0.877 | 0.867 | 0.841 | 0.863 | 0.871 | 0.818 | 0.867 | 0.889 | 0.886 | 0.896 | 0.890 | 0.834 | 0.885 | |
| Chills | 0.803 | 0.801 | 0.921 | 0.944 | 1.000 | 0.803 | 0.777 | 0.775 | 0.786 | 0.793 | 0.761 | 0.778 | 0.808 | 0.767 | 0.811 | 0.815 | 0.788 | 0.805 | |
| Joint Pain | 0.841 | 0.815 | 0.975 | 0.969 | 1.000 | 0.836 | 0.821 | 0.807 | 0.822 | 0.829 | 0.806 | 0.819 | 0.845 | 0.842 | 0.850 | 0.844 | 0.838 | 0.850 | |
| Muscle Pain | 0.827 | 0.802 | 0.958 | 0.918 | 1.000 | 0.824 | 0.811 | 0.787 | 0.808 | 0.817 | 0.803 | 0.813 | 0.829 | 0.785 | 0.815 | 0.816 | 0.793 | 0.831 | |
| Local side effects | 0.805 | 0.762 | 0.883 | 0.922 | 1.000 | 0.811 | 0.796 | 0.763 | 0.791 | 0.804 | 0.772 | 0.797 | 0.802 | 0.768 | 0.807 | 0.823 | 0.791 | 0.801 | |
| Sputnik V | Fever | 0.923 | 0.885 | 0.975 | 0.977 | 1.000 | 0.923 | 0.906 | 0.870 | 0.902 | 0.912 | 0.833 | 0.906 | 0.860 | 0.867 | 0.867 | 0.873 | 0.813 | 0.857 |
| Fatigue | 0.907 | 0.883 | 0.959 | 0.966 | 1.000 | 0.912 | 0.903 | 0.874 | 0.890 | 0.903 | 0.858 | 0.902 | 0.890 | 0.873 | 0.881 | 0.895 | 0.872 | 0.887 | |
| Headache | 0.905 | 0.874 | 0.975 | 0.960 | 1.000 | 0.910 | 0.901 | 0.861 | 0.902 | 0.907 | 0.865 | 0.901 | 0.902 | 0.859 | 0.889 | 0.904 | 0.850 | 0.892 | |
| Nausea | 0.888 | 0.860 | 0.901 | 0.985 | 1.000 | 0.869 | 0.857 | 0.816 | 0.844 | 0.868 | 0.659 | 0.857 | 0.842 | 0.806 | 0.845 | 0.813 | 0.688 | 0.843 | |
| Chills | 0.889 | 0.838 | 0.994 | 0.990 | 1.000 | 0.882 | 0.861 | 0.808 | 0.841 | 0.859 | 0.728 | 0.860 | 0.794 | 0.769 | 0.774 | 0.793 | 0.682 | 0.759 | |
| Joint Pain | 0.908 | 0.856 | 0.916 | 0.967 | 1.000 | 0.901 | 0.887 | 0.832 | 0.879 | 0.892 | 0.825 | 0.889 | 0.895 | 0.832 | 0.874 | 0.892 | 0.826 | 0.909 | |
| Muscle Pain | 0.901 | 0.873 | 0.951 | 0.936 | 1.000 | 0.905 | 0.895 | 0.865 | 0.882 | 0.897 | 0.841 | 0.895 | 0.910 | 0.881 | 0.904 | 0.909 | 0.852 | 0.907 | |
| Local side effects | 0.899 | 0.874 | 0.960 | 0.955 | 1.000 | 0.910 | 0.894 | 0.874 | 0.893 | 0.901 | 0.869 | 0.896 | 0.893 | 0.841 | 0.881 | 0.892 | 0.875 | 0.888 | |
| BBIBP-CorV | Fever | 0.837 | 0.792 | 0.943 | 0.939 | 1.000 | 0.847 | 0.814 | 0.789 | 0.826 | 0.827 | 0.802 | 0.816 | 0.831 | 0.820 | 0.837 | 0.847 | 0.809 | 0.828 |
| Fatigue | 0.848 | 0.814 | 0.919 | 0.938 | 1.000 | 0.843 | 0.838 | 0.805 | 0.837 | 0.842 | 0.824 | 0.838 | 0.840 | 0.803 | 0.845 | 0.850 | 0.838 | 0.841 | |
| Headache | 0.860 | 0.828 | 0.953 | 0.934 | 1.000 | 0.857 | 0.848 | 0.828 | 0.842 | 0.853 | 0.827 | 0.849 | 0.849 | 0.834 | 0.856 | 0.858 | 0.841 | 0.848 | |
| Nausea | 0.844 | 0.794 | 0.989 | 0.969 | 1.000 | 0.841 | 0.819 | 0.777 | 0.815 | 0.830 | 0.780 | 0.826 | 0.829 | 0.810 | 0.836 | 0.853 | 0.802 | 0.823 | |
| Chills | 0.816 | 0.783 | 0.982 | 0.913 | 1.000 | 0.817 | 0.795 | 0.781 | 0.811 | 0.812 | 0.794 | 0.800 | 0.796 | 0.782 | 0.818 | 0.810 | 0.787 | 0.791 | |
| Joint Pain | 0.860 | 0.821 | 0.967 | 0.947 | 1.000 | 0.863 | 0.847 | 0.817 | 0.849 | 0.855 | 0.833 | 0.847 | 0.855 | 0.822 | 0.857 | 0.860 | 0.848 | 0.854 | |
| Muscle Pain | 0.841 | 0.817 | 0.918 | 0.938 | 1.000 | 0.857 | 0.837 | 0.814 | 0.836 | 0.843 | 0.827 | 0.838 | 0.842 | 0.812 | 0.844 | 0.856 | 0.841 | 0.840 | |
| Local side effects | 0.853 | 0.818 | 0.948 | 0.931 | 1.000 | 0.852 | 0.840 | 0.816 | 0.844 | 0.847 | 0.822 | 0.842 | 0.817 | 0.782 | 0.829 | 0.833 | 0.820 | 0.821 | |
| COVAXIN | Fever | 0.917 | 0.884 | 0.854 | 0.973 | 1.000 | 0.863 | 0.844 | 0.826 | 0.828 | 0.849 | 0.810 | 0.834 | – | – | – | – | – | – |
| Fatigue | 0.885 | 0.898 | 0.946 | 0.983 | 0.875 | 0.869 | 0.844 | 0.834 | 0.838 | 0.851 | 0.813 | 0.844 | – | – | – | – | – | – | |
| Headache | 0.874 | 0.879 | 0.972 | 0.993 | 1.000 | 0.855 | 0.842 | 0.817 | 0.836 | 0.850 | 0.811 | 0.834 | – | – | – | – | – | – | |
| Nausea | 0.951 | 0.929 | 0.870 | 0.999 | 0.925 | 0.876 | 0.832 | 0.820 | 0.823 | 0.853 | 0.782 | 0.811 | – | – | – | – | – | – | |
| Chills | 0.873 | 0.876 | 0.931 | 0.988 | 1.000 | 0.843 | 0.788 | 0.772 | 0.784 | 0.805 | 0.682 | 0.783 | – | – | – | – | – | – | |
| Joint Pain | 0.899 | 0.894 | 0.913 | 0.987 | 1.000 | 0.879 | 0.875 | 0.849 | 0.869 | 0.873 | 0.839 | 0.873 | – | – | – | – | – | – | |
| Muscle Pain | 0.888 | 0.888 | 0.974 | 0.955 | 0.879 | 0.867 | 0.850 | 0.842 | 0.842 | 0.847 | 0.817 | 0.847 | – | – | – | – | – | – | |
| Local side effects | 0.861 | 0.859 | 1.000 | 0.988 | 0.861 | 0.838 | 0.788 | 0.781 | 0.805 | 0.816 | 0.781 | 0.790 | – | – | – | – | – | – | |
| BNT162b2 | Fever | 0.820 | 0.824 | 0.794 | 0.960 | 0.717 | 0.660 | 0.582 | 0.577 | 0.579 | 0.552 | 0.504 | 0.567 | – | – | – | – | – | – |
| Fatigue | 0.836 | 0.862 | 0.851 | 0.988 | 0.726 | 0.701 | 0.680 | 0.660 | 0.716 | 0.702 | 0.547 | 0.651 | – | – | – | – | – | – | |
| Headache | 0.814 | 0.809 | 0.905 | 0.996 | 0.702 | 0.692 | 0.652 | 0.662 | 0.605 | 0.605 | 0.562 | 0.594 | – | – | – | – | – | – | |
| Nausea | 0.875 | 0.809 | 0.874 | 0.978 | 0.724 | 0.670 | 0.675 | 0.655 | 0.616 | 0.636 | 0.553 | 0.615 | – | – | – | – | – | – | |
| Chills | 0.804 | 0.794 | 0.594 | 0.995 | 1.000 | 0.563 | 0.486 | 0.482 | 0.505 | 0.520 | 0.549 | 0.506 | – | – | – | – | – | – | |
| Joint Pain | 0.868 | 0.843 | 0.797 | 0.997 | 1.000 | 0.728 | 0.672 | 0.674 | 0.643 | 0.660 | 0.631 | 0.635 | – | – | – | – | – | – | |
| Muscle Pain | 0.864 | 0.821 | 0.988 | 0.978 | 0.718 | 0.735 | 0.630 | 0.637 | 0.638 | 0.648 | 0.594 | 0.612 | – | – | – | – | – | – | |
| Local side effects | 0.871 | 0.828 | 0.993 | 0.994 | 0.786 | 0.649 | 0.664 | 0.535 | 0.707 | 0.725 | 0.693 | 0.622 | – | – | – | – | – | – | |
| mRNA-1273 | Fever | 0.809 | 0.819 | 0.883 | 0.975 | 1.000 | 0.722 | 0.683 | 0.622 | 0.689 | 0.709 | 0.571 | 0.689 | – | – | – | – | – | – |
| Fatigue | 0.792 | 0.796 | 0.827 | 0.954 | 1.000 | 0.752 | 0.717 | 0.711 | 0.763 | 0.762 | 0.657 | 0.701 | – | – | – | – | – | – | |
| Headache | 0.830 | 0.796 | 0.978 | 0.997 | 0.774 | 0.758 | 0.744 | 0.737 | 0.735 | 0.785 | 0.706 | 0.748 | – | – | – | – | – | – | |
| Nausea | 0.913 | 0.904 | 0.905 | 0.997 | 0.840 | 0.763 | 0.715 | 0.649 | 0.787 | 0.760 | 0.643 | 0.725 | – | – | – | – | – | – | |
| Chills | 0.806 | 0.827 | 0.857 | 0.999 | 1.000 | 0.702 | 0.721 | 0.662 | 0.648 | 0.698 | 0.615 | 0.690 | – | – | – | – | – | – | |
| Joint Pain | 0.814 | 0.855 | 0.979 | 0.979 | 1.000 | 0.791 | 0.734 | 0.684 | 0.785 | 0.774 | 0.708 | 0.731 | – | – | – | – | – | – | |
| Muscle Pain | 0.865 | 0.849 | 0.856 | 0.986 | 0.782 | 0.754 | 0.776 | 0.742 | 0.759 | 0.771 | 0.709 | 0.717 | – | – | – | – | – | – | |
| Local side effects | 0.900 | 0.879 | 1.000 | 0.982 | 0.842 | 0.835 | 0.850 | 0.815 | 0.771 | 0.824 | 0.773 | 0.810 | – | – | – | – | – | – | |
Model average performance has been calculated on training, validation, and test set using AUC-ROC parameter.
Second Dose Model Performance.xlsx.