27 lines
629 B
ObjectPascal
27 lines
629 B
ObjectPascal
uses MLABC;
|
|
uses TestHelpers in '..\TestHelpers.pas';
|
|
|
|
begin
|
|
var ds := Datasets.Iris;
|
|
var df := ds.Data;
|
|
|
|
var pipe :=
|
|
DataPipeline.BuildClassification(
|
|
ds.Target,
|
|
ds.Features,
|
|
new StandardScaler,
|
|
new LogisticRegression
|
|
);
|
|
|
|
var (trainDf, testDf) := df.TrainTestSplit(0.2, seed := 3);
|
|
pipe.Fit(trainDf);
|
|
|
|
var classes := pipe.GetClassLabels;
|
|
var proba := pipe.PredictProba(testDf);
|
|
|
|
Check(proba.RowCount = testDf.RowCount, 'Probability row count mismatch');
|
|
Check(proba.ColCount = classes.Length, 'Probability class count mismatch');
|
|
CheckProbabilityRowsSumToOne(proba);
|
|
end.
|
|
|