Exemplo n.º 1
0
def CompareDataPerformance(df, names, ptitle="", ModelIndex=0, Epochs=10):
    print("Run for: ", ptitle)
    Train, Test = pf.SplitData(df)
    XTrain, YTrain = pf.TrainandTarget(Train, names)
    XTest, YTest = pf.TrainandTarget(Test, names)
    p_model, p_history = tam.BuildTrainSeqModel(ModelIndex, Epochs, XTrain,
                                                YTrain, XTest, YTest)
    plots = [
        pld.PlotPerformance(p_history.history, ptitle + str(ModelIndex)),
        pld.ROCCurve(XTest, YTest, p_model, ptitle + str(ModelIndex))
    ]
    pld.SavePlots(plots, ptitle + "MI" + str(ModelIndex) + ".pdf")