Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- public BikeSharingDemandPrediction Predict(BikeSharingDemandSample sample)
- {
- return _predictionEngine.Predict(sample);
- }
- public RegressionMetrics Evaluate(string testDataLocation)
- {
- var testDataView = _mlContext.Data.LoadFromTextFile<BikeSharingDemandSample>(
- path: testDataLocation,
- hasHeader: true,
- separatorChar: ',',
- allowQuoting: true,
- allowSparse: false);
- var predictions = _trainedModel.Transform(testDataView);
- return _mlContext.Regression.Evaluate(predictions, "Label", "Score");
- }
- public void SaveModel()
- {
- _mlContext.Model.Save(_trainedModel, _trainingDataView.Schema, "./BikeSharingDemandsModel.zip");
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement