在go语言中实现马尔科夫链预测下一状态概率的示例
Go / 管理员 发布于 3星期前 69
马尔可夫链在很多情况下都很有用，尤其是在交易中,它可以用来预测下一个趋势是看跌、看涨还是停滞。
如果你不知道这些代码想要做什么，可以自行先了解一下马尔可夫链。
第1部分示例代码 ：
package main
import (
"fmt"
"strconv"
"strings"
"time"
"math/rand"
"github.com/gonum/matrix/mat64"
)
// 状态空间
var states = []string{"sleep", "icecream", "run"}
// 可能发生的事件
var transitionFromSleep = []string{
"SS",
"SR",
"SI",
}
var transitionFromRun = []string{
"RS",
"RR",
"RI",
}
var transitionFromIceCream = []string{
"IS",
"IR",
"II",
}
var transitionMatrix mat64.Matrix
func main() {
// 概率矩阵（过渡矩阵）
// 数字是凭空捏造的...
transitionMatrixString := "[0.2 0.6 0.2;0.1 0.6 0.3; 0.2 0.7 0.1]"
transitionMatrix = matrix(transitionMatrixString)
// 打印所有过渡矩阵元素
fmt.Printf("transitionMatrix :\n%v\n\n", mat64.Formatted(transitionMatrix, mat64.Prefix(""), mat64.Excerpt(0)))
// 执行简单的正确性检查。
// 所有概率之和必须为 1，因此我们有 3 行，最终结果必须等于 3。
// 结果必须等于 3
s := mat64.Sum(transitionMatrix)
s1 := fmt.Sprintf("%0.1f", s)
if s1 != "3.0" {
panic("概率总和必须等于 3，每行总和为 1。检查您的转换矩阵")
} else {
fmt.Println("过渡矩阵检查完毕，进入下一阶段！")
}
activityForecast(2)
}
func matrixByRow(a mat64.Matrix, rowNumber int) []float64 {
dst := []float64{0.0, 0.0, 0.0}
mat64.Row(dst, rowNumber, a)
return dst
}
func sampleSet(cdf []float64) int {
// https://stackoverflow.com/questions/50507513/golang-choice-number-from-slice-array-with-given-probability
rand.Seed(time.Now().UnixNano())
r := rand.Float64()
bucket := 0
for r > cdf[bucket] {
bucket++
if bucket == len(cdf) {
bucket--
break // 需要这样做来防止运行时出错：索引超出范围
}
}
return bucket
}
func activityForecast(days int) {
// 默认状态
todayActivity := "sleep"
fmt.Println("Start state : ", todayActivity)
activityList := []string{}
activityList = append(activityList, todayActivity)
i := 0
// 计算活动列表的概率
prob := 1.0
for i < days {
if todayActivity == "sleep" {
sleepNextProbabilityDistribution := matrixByRow(transitionMatrix, 0)
randomChoice := sampleSet(sleepNextProbabilityDistribution)
change := transitionFromSleep[randomChoice]
if change == "SS" {
prob = prob * 0.2
activityList = append(activityList, "sleep")
} else if change == "SR" {
prob = prob * 0.6
todayActivity = "run"
activityList = append(activityList, "run")
} else {
prob = prob * 0.2
todayActivity = "icecream"
activityList = append(activityList, "icecream")
}
} else if todayActivity == "run" {
runNextProbabilityDistribution := matrixByRow(transitionMatrix, 1)
randomChoice := sampleSet(runNextProbabilityDistribution)
change := transitionFromRun[randomChoice]
if change == "RR" {
prob = prob * 0.5
activityList = append(activityList, "run")
} else if change == "RS" {
prob = prob * 0.2
todayActivity = "sleep"
activityList = append(activityList, "sleep")
} else {
prob = prob * 0.3
todayActivity = "icecream"
activityList = append(activityList, "icecream")
}
} else {
sleepNextProbabilityDistribution := matrixByRow(transitionMatrix, 2)
randomChoice := sampleSet(sleepNextProbabilityDistribution)
change := transitionFromIceCream[randomChoice]
if change == "II" {
prob = prob * 0.1
activityList = append(activityList, "icecream")
} else if change == "IS" {
prob = prob * 0.2
todayActivity = "sleep"
activityList = append(activityList, "sleep")
} else {
prob = prob * 0.7
todayActivity = "run"
activityList = append(activityList, "run")
}
}
i = i + 1
}
fmt.Println("Possible states : ", activityList)
fmt.Println("End state after " + strconv.Itoa(days) + " days : " + todayActivity)
fmt.Println("可能的状态序列概率：", strconv.FormatFloat(prob, 'f', 6, 64))
}
// 从仿真 NumPy 创建矩阵的方法示例：https://www.zongscan.com/demo333/96244.html
// 警告：此函数不对输入字符串进行错误检查。
// 您可能需要修改该函数，以便对以下情况进行正确性检查：
// 例如只有一对 [ ]
// 所有列和行都有数字或 0
// 所有行的大小相同 [pynum 将抛出 "ValueError: Rows not the same size.］
func matrix(str string) *mat64.Dense {
// 删除 [ 和 ]
str = strings.Replace(str, "[", "", -1)
str = strings.Replace(str, "]", "", -1)
// 计算总行数
parts := strings.SplitN(str, ";", -1)
rows := len(parts)
// 计算列总数
colSlice := strings.Fields(parts[0])
columns := len(colSlice)
// 将所有 ; 替换为空格
str = strings.Replace(str, ";", " ", -1)
// 将字符串转换为片段
// 摘自 https://www.zongscan.com/demo333/96243.html
elements := strings.Fields(str)
//为新矩阵填充数据（密集型）
data := make([]float64, rows*columns)
for i := range data {
floatValue, _ := strconv.ParseFloat(elements[i], 64)
data[i] = floatValue
}
M := mat64.NewDense(rows, columns, data)
return M
}
输出示例：
transitionMatrix ：
⎡0.2 0.6 0.2⎤
⎢0.1 0.6 0.3⎥
⎣0.2 0.7 0.1⎦
过渡矩阵检查完毕，进入下一阶段！
开始状态：睡眠
可能的状态 [睡眠 运行 运行］
2 天后的结束状态：运行
可能的状态序列概率 ： 0.300000
第2部分示例代码 ：
package main
import (
"fmt"
"strconv"
"strings"
"time"
"math/rand"
"github.com/gonum/matrix/mat64"
)
// 状态空间
var states = []string{"sleep", "icecream", "run"}
// 可能发生的事件
var transitionFromSleep = []string{
"SS",
"SR",
"SI",
}
var transitionFromRun = []string{
"RS",
"RR",
"RI",
}
var transitionFromIceCream = []string{
"IS",
"IR",
"II",
}
var transitionMatrix mat64.Matrix
func main() {
// 概率矩阵（过渡矩阵）
// 数字是凭空捏造的...
transitionMatrixString := "[0.2 0.6 0.2;0.1 0.6 0.3; 0.2 0.7 0.1]"
transitionMatrix = matrix(transitionMatrixString)
// 打印所有过渡矩阵元素
fmt.Printf("transitionMatrix :\n%v\n\n", mat64.Formatted(transitionMatrix, mat64.Prefix(""), mat64.Excerpt(0)))
// 执行简单的正确性检查。
// 所有概率之和必须为 1，因此我们有 3 行，最终结果必须等于 3。
// 结果必须等于 3
s := mat64.Sum(transitionMatrix)
s1 := fmt.Sprintf("%0.1f", s)
if s1 != "3.0" {
panic("概率总和必须等于 3，每行总和为 1。检查您的过渡矩阵")
} else {
fmt.Println("过渡矩阵已检查完毕，进入下一阶段！")
}
// 保存每个活动列表
activityListSlice := [][]string{}
count := 0
numberOfIterations := 10000
// 迭代
for i := 1; i < numberOfIterations; i++ {
activityListSlice = append(activityListSlice, activityForecast(2))
}
// 查看我们收集的所有 "活动列表
//fmt.Println(activityListSlice)
// 遍历列表，以获得以状态："运行 "结束的所有活动的计数
for _, v := range activityListSlice {
if v[2] == "run" {
count++
}
}
// 计算从状态 "睡眠 "开始到状态 "运行 "结束的概率： "睡眠 "状态，并以 "运行 "状态结束
inner := float64(count) / float64(numberOfIterations)
percentage := inner * 100.00
fmt.Printf("开始于状态 "睡眠 "并结束于状态 "睡眠 "的概率 睡眠 "状态和 "运行 "状态的概率 : %0.2f%%\n", percentage)
}
func matrixByRow(a mat64.Matrix, rowNumber int) []float64 {
dst := []float64{0.0, 0.0, 0.0}
mat64.Row(dst, rowNumber, a)
return dst
}
func sampleSet(cdf []float64) int {
// https://stackoverflow.com/questions/50507513/golang-choice-number-from-slice-array-with-given-probability
rand.Seed(time.Now().UnixNano())
r := rand.Float64()
bucket := 0
for r > cdf[bucket] {
bucket++
if bucket == len(cdf) {
bucket--
break // 需要这样做来防止运行时出错：索引超出范围
}
}
return bucket
}
func activityForecast(days int) []string {
// 默认状态
todayActivity := "sleep"
activityList := []string{}
activityList = append(activityList, todayActivity)
i := 0
// 计算活动列表的概率
prob := 1.0
for i < days {
if todayActivity == "sleep" {
sleepNextProbabilityDistribution := matrixByRow(transitionMatrix, 0)
randomChoice := sampleSet(sleepNextProbabilityDistribution)
change := transitionFromSleep[randomChoice]
if change == "SS" {
prob = prob * 0.2
activityList = append(activityList, "sleep")
} else if change == "SR" {
prob = prob * 0.6
todayActivity = "run"
activityList = append(activityList, "run")
} else {
prob = prob * 0.2
todayActivity = "icecream"
activityList = append(activityList, "icecream")
}
} else if todayActivity == "run" {
runNextProbabilityDistribution := matrixByRow(transitionMatrix, 1)
randomChoice := sampleSet(runNextProbabilityDistribution)
change := transitionFromRun[randomChoice]
if change == "RR" {
prob = prob * 0.5
activityList = append(activityList, "run")
} else if change == "RS" {
prob = prob * 0.2
todayActivity = "sleep"
activityList = append(activityList, "sleep")
} else {
prob = prob * 0.3
todayActivity = "icecream"
activityList = append(activityList, "icecream")
}
} else {
sleepNextProbabilityDistribution := matrixByRow(transitionMatrix, 2)
randomChoice := sampleSet(sleepNextProbabilityDistribution)
change := transitionFromIceCream[randomChoice]
if change == "II" {
prob = prob * 0.1
activityList = append(activityList, "icecream")
} else if change == "IS" {
prob = prob * 0.2
todayActivity = "sleep"
activityList = append(activityList, "sleep")
} else {
prob = prob * 0.7
todayActivity = "run"
activityList = append(activityList, "run")
}
}
i++
}
return activityList
}
// 从仿真 NumPy 创建矩阵的方法示例：https://www.zongscan.com/demo333/96244.html
// 警告：此函数不对输入字符串进行错误检查。
// 您可能需要修改该函数，以便对以下情况进行正确性检查：
// 例如只有一对 [ ]
// 所有列和行都有数字或 0
// 所有行的大小相同 [pynum 将抛出 "ValueError: Rows not the same size.］
func matrix(str string) *mat64.Dense {
// 删除 [ 和 ]
str = strings.Replace(str, "[", "", -1)
str = strings.Replace(str, "]", "", -1)
// 计算总行数
parts := strings.SplitN(str, ";", -1)
rows := len(parts)
// 计算列总数
colSlice := strings.Fields(parts[0])
columns := len(colSlice)
// 将所有 ; 替换为空格
str = strings.Replace(str, ";", " ", -1)
// 将字符串转换为片段
// 摘自 https://www.zongscan.com/demo333/96243.html
elements := strings.Fields(str)
// 为新矩阵填充数据（密集型）
data := make([]float64, rows*columns)
for i := range data {
floatValue, _ := strconv.ParseFloat(elements[i], 64)
data[i] = floatValue
}
M := mat64.NewDense(rows, columns, data)
return M
}输出示例：
transitionMatrix ：
⎡0.2 0.6 0.2⎤
⎢0.1 0.6 0.3⎥
⎣0.2 0.7 0.1⎦
过渡矩阵检查完毕，进入下一阶段！
开始于状态 "睡眠 "并结束于状态 "睡眠 "的概率 睡眠 "状态和 "运行 "状态的概率 : 47.54- 相关文章
- 在go语言中实现原子操作示例代码(0个评论)
- 在go语言中实现蒙提霍尔（三门问题）示例代码(0个评论)
- 在go语言种获取进程ID后面的可执行文件名示例代码(0个评论)
- 在go语言如何确定时间变量是否具有相同的日历日？(0个评论)
- 在go语言中使用PGX CopyFrom扩展包实现向Postgres数据库插入数据示例(1个评论)
- 近期文章
- 在ubuntu系统中实现mysql数据存储目录迁移流程步骤(0个评论)
- 在mysql中使用存储过程批量新增测试数据流程步骤(0个评论)
- docker + gitea搭建一个git服务器流程步骤(0个评论)
- Laravel 10.34版本发布(0个评论)
- PHP8中的JIT编译器可以将代码性能提高20%以上(0个评论)
- 在php语言中实现网站ssl证书过期时间检查功能示例代码(0个评论)
- 在php语言中实现谷歌登陆功能：google授权api示例代码(0个评论)
- 在php语言中实现文件加密功能示例(0个评论)
- 以优雅的方式编写应用程序接口集成和 SDK(0个评论)
- 在php+laravel项目中使用Saloon扩展包编写API集成(0个评论)
- 近期评论
-
博主 在
佛跳墙vpn软件不会用?上不了网?佛跳墙vpn常见问题以及解决办法中评论 @1111老铁这个不行了，可以看看近期评论的其他文章.. -
1111 在
佛跳墙vpn软件不会用?上不了网?佛跳墙vpn常见问题以及解决办法中评论 网站不能打开，博主百忙中能否发个APP下载链接，佛跳墙或极光.. -
路人 在
php中使用hyperf框架调用讯飞星火大模型实现国内版chatgpt功能示例中评论 教程很详细，如果加个前端chatgpt对话页面就完美了.. -
博主 在
科学上网翻墙之v2rayN-Core客户端免费公益节点使用教程中评论 @ mashrdn 多切换几个节点测试，免费ssr是没那么稳.. -
mashrdn 在
科学上网翻墙之v2rayN-Core客户端免费公益节点使用教程中评论 V2rayn免费节点添加上去了，youtobe无法打开网页，是怎么回事..
Copyright·© 2019 侯体宗版权所有·
粤ICP备20027696号
