BotSharp/BotSharp.Algorithm/Estimators/Lidstone.cs
2018-09-09 22:56:32 -05:00

67 lines
2.1 KiB
C#
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

/*
* BotSharp.Algorithm
* Copyright (C) 2018 Haiping Chen
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
using BotSharp.Algorithm.Statistics;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
namespace BotSharp.Algorithm.Estimators
{
/// <summary>
/// Lidstone smoothing is a technique used to smooth categorical data.
/// In statistics, it's called additive smoothing or Laplace smoothing.
/// Given an observation x = (x1, …, xd) from a multinomial distribution with N trials, a "smoothed" version of the data gives the estimator.
/// https://en.wikipedia.org/wiki/Additive_smoothing
/// Used as Multinomial Naive Bayes
/// </summary>
public class Lidstone : IEstimator
{
/// <summary>
/// α > 0 is the smoothing parameter
/// </summary>
public double Alpha { get; set; }
/// <summary>
/// Probability
/// </summary>
/// <param name="dist">distribution</param>
/// <param name="sample">sample value</param>
/// <returns></returns>
public double Prob(List<Probability> dist, string sample)
{
if(Alpha == 0)
{
Alpha = 0.5D;
}
// observation x = (x1, ..., xd)
var p = dist.Find(f => f.Value == sample);
int x = p == null ? 0 : p.Freq;
// N trials
int _N = dist.Sum(f => f.Freq);
int _d = dist.Count;
return (x + Alpha) / (_N + Alpha * _d);
}
}
}