from a_utils import *
from a7_corr_svm import correction_svm
team_file = getOldDataset()
print("A7./ 70:30 Chéo, SVM to Predict")
correction_svm(team_file.train, team_file.resultColName,
                  team_file.listFileTest, 1, 0.3, IF_Method.UnivariateSelection, 25)
correction_svm(team_file.train, team_file.resultColName,
                  team_file.listFileTest, 1, 0.3, IF_Method.PearsonCorrelationMatrix, 25)
correction_svm(team_file.train, team_file.resultColName,
                  team_file.listFileTest, 1, 0.3, IF_Method.FeatureImportance, 25)

from a_utils import *
from a7_corr_svm import correction_svm
team_file = getOldDataset()
print("A7./ 70:30 Chéo, SVM to Predict")
correction_svm(team_file.train, team_file.resultColName,
                  team_file.listFileTest, 1, 0.3, IF_Method.UnivariateSelection, 25)
correction_svm(team_file.train, team_file.resultColName,
                  team_file.listFileTest, 1, 0.3, IF_Method.PearsonCorrelationMatrix, 25)
correction_svm(team_file.train, team_file.resultColName,
                  team_file.listFileTest, 1, 0.3, IF_Method.FeatureImportance, 25)
