public void TestIDFModel() { string expectedInputCol = "rawFeatures"; string expectedOutputCol = "features"; int expectedDocFrequency = 100; IDF idf = new IDF() .SetInputCol(expectedInputCol) .SetOutputCol(expectedOutputCol) .SetMinDocFreq(expectedDocFrequency); Assert.Equal(expectedInputCol, idf.GetInputCol()); Assert.Equal(expectedOutputCol, idf.GetOutputCol()); Assert.Equal(expectedDocFrequency, idf.GetMinDocFreq()); using (var tempDirectory = new TemporaryDirectory()) { string savePath = Path.Join(tempDirectory.Path, "IDF"); idf.Save(savePath); IDF loadedIdf = IDF.Load(savePath); Assert.Equal(idf.Uid(), loadedIdf.Uid()); } }