MCPcopy Create free account

hub / github.com/Leavesfly/TinyDL-0.01 / functions

Functions1,708 in github.com/Leavesfly/TinyDL-0.01

↓ 2 callersMethoddownloadMnist
()
src/main/java/io/leavesfly/tinydl/mlearning/dataset/simple/MnistDataSet.java:106
↓ 2 callersMethodequals
优化的equals方法,比较两个NdArray对象是否相等 @param obj 另一个对象 @return 是否相等
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1746
↓ 2 callersMethodescapeJsonString
转义JSON字符串 @param str 原始字符串 @return 转义后的字符串
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfoExporter.java:335
↓ 2 callersMethodevaluateAgent
评估智能体性能 @param agent 智能体 @param env 环境 @param numEvaluationEpisodes 评估回合数
src/main/java/io/leavesfly/tinydl/example/rl/CartPoleDQNExample.java:156
↓ 2 callersMethodevaluateAgent
评估智能体性能 @param agent 智能体 @param env 环境 @param numEvaluationEpisodes 评估回合数
src/main/java/io/leavesfly/tinydl/example/rl/GridWorldREINFORCEExample.java:176
↓ 2 callersMethodexportSimpleReport
生成简化的JSON报告 @param model 模型 @param filePath 保存路径
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfoExporter.java:206
↓ 2 callersMethodfillAll
用指定值填充整个数组 @param number 填充值
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1559
↓ 2 callersMethodfindBestAction
找到Q值最大的动作索引 @param qValues Q值数组 @return 最优动作索引
src/main/java/io/leavesfly/tinydl/modality/rl/policy/EpsilonGreedyPolicy.java:131
↓ 2 callersMethodforward
(NdArray... inputs)
src/main/java/io/leavesfly/tinydl/nnet/layer/norm/BatchNorm.java:102
↓ 2 callersMethodforward
前向传播计算加法 执行两个NdArray的加法运算。如果两个输入的形状不同, 则对第二个输入进行广播以匹配第一个输入的形状。 @param inputs 输入的NdArray数组,长度为2 @return 加法运算结果的NdArray
src/main/java/io/leavesfly/tinydl/func/base/Add.java:29
↓ 2 callersMethodforward
前向传播计算转置 计算输入数组的转置。 @param inputs 输入的NdArray数组,长度为1 @return 转置后的NdArray
src/main/java/io/leavesfly/tinydl/func/matrix/Transpose.java:23
↓ 2 callersMethodforward
前向传播计算重塑 将输入数组重塑为指定形状。 @param inputs 输入的NdArray数组,长度为1 @return 重塑后的NdArray
src/main/java/io/leavesfly/tinydl/func/matrix/Reshape.java:37
↓ 2 callersMethodforward
前向传播计算线性变换 执行线性变换操作:y = x w + b,其中b为可选偏置项。 @param inputs 输入的NdArray数组,长度为2或3(x, w, [b]) @return 线性变换后的NdArray
src/main/java/io/leavesfly/tinydl/func/matrix/Linear.java:24
↓ 2 callersMethodforward
前向传播计算SoftMax 计算SoftMax函数值:softmax(x_i) = e^(x_i) / Σ(e^(x_j)) @param inputs 输入的NdArray数组,长度为1 @return SoftMax函数值的NdArray
src/main/java/io/leavesfly/tinydl/func/matrix/SoftMax.java:25
↓ 2 callersMethodforward
前向传播计算矩阵乘法 计算两个矩阵的内积(点积):x w @param inputs 输入的NdArray数组,长度为2 @return 矩阵乘法结果的NdArray
src/main/java/io/leavesfly/tinydl/func/matrix/MatMul.java:23
↓ 2 callersMethodforward
前向传播计算求和 计算输入数组所有元素的和。 @param inputs 输入的NdArray数组,长度为1 @return 求和结果的NdArray
src/main/java/io/leavesfly/tinydl/func/matrix/Sum.java:26
↓ 2 callersMethodforward
前向传播计算最大值 计算输入数组沿指定轴的最大值。 @param inputs 输入的NdArray数组,长度为1 @return 最大值的NdArray
src/main/java/io/leavesfly/tinydl/func/math/Max.java:37
↓ 2 callersMethodforward
前向传播计算指数 计算输入值的指数值:e^x @param inputs 输入的NdArray数组,长度为1 @return 指数值的NdArray
src/main/java/io/leavesfly/tinydl/func/math/Exp.java:23
↓ 2 callersMethodforward
前向传播计算ReLU 计算ReLU函数值:max(0, x) @param inputs 输入的NdArray数组,长度为1 @return ReLU函数值的NdArray
src/main/java/io/leavesfly/tinydl/func/math/ReLu.java:24
↓ 2 callersMethodforward
前向传播计算对数 计算输入值的对数值:ln(x) @param inputs 输入的NdArray数组,长度为1 @return 对数值的NdArray
src/main/java/io/leavesfly/tinydl/func/math/Log.java:23
↓ 2 callersMethodgetAction
获取执行的动作 @return 执行的动作
src/main/java/io/leavesfly/tinydl/modality/rl/Experience.java:92
↓ 2 callersMethodgetAgentPosition
获取智能体当前位置 @return 包含x和y坐标的映射
src/main/java/io/leavesfly/tinydl/modality/rl/environment/GridWorldEnvironment.java:365
↓ 2 callersMethodgetAllLayersExpertUsageRates
获取各层专家使用率的汇总统计 @return 每层的专家使用率数组
src/main/java/io/leavesfly/tinydl/modality/nlp/MoEGPTModel.java:402
↓ 2 callersMethodgetAllUCBValues
获取所有臂的UCB值 @return UCB值数组
src/main/java/io/leavesfly/tinydl/modality/rl/agent/UCBBanditAgent.java:122
↓ 2 callersMethodgetBatchSize
获取批次大小 @return 批次大小
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:384
↓ 2 callersMethodgetCreatedTime
获取创建时间 @return 创建时间
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:138
↓ 2 callersMethodgetEmbedSize
获取嵌入维度 @return 嵌入维度
src/main/java/io/leavesfly/tinydl/nnet/layer/embedd/Embedding.java:149
↓ 2 callersMethodgetExpertUsageRates
获取专家使用率统计信息 @return 专家使用率数组
src/main/java/io/leavesfly/tinydl/modality/nlp/layer/MoELayer.java:303
↓ 2 callersMethodgetFrameworkVersion
获取框架版本 @return 框架版本
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:121
↓ 2 callersMethodgetGoalPosition
获取目标位置 @return 包含x和y坐标的映射
src/main/java/io/leavesfly/tinydl/modality/rl/environment/GridWorldEnvironment.java:377
↓ 2 callersMethodgetLastModifiedTime
获取最后修改时间 @return 最后修改时间
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:154
↓ 2 callersMethodgetLearningRate
获取学习率 @return 学习率
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:367
↓ 2 callersMethodgetLossFunction
获取损失函数 @return 损失函数
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:401
↓ 2 callersMethodgetOptimizerType
获取优化器类型 @return 优化器类型
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:350
↓ 2 callersMethodgetParameterCount
计算模型的总参数量 @return 模型总参数量
src/main/java/io/leavesfly/tinydl/modality/nlp/GPT2Model.java:300
↓ 2 callersMethodgetParameterStats
获取参数统计信息 @param parameters 参数映射 @return 统计信息
src/main/java/io/leavesfly/tinydl/mlearning/ParameterManager.java:181
↓ 2 callersMethodgetPosteriorStdDev
获取指定臂的后验标准差 @param armIndex 臂索引 @return 后验标准差
src/main/java/io/leavesfly/tinydl/modality/rl/agent/ThompsonSamplingBanditAgent.java:187
↓ 2 callersMethodgetQValues
获取状态的所有动作Q值 @param state 状态 @return Q值向量
src/main/java/io/leavesfly/tinydl/modality/rl/policy/EpsilonGreedyPolicy.java:204
↓ 2 callersMethodgetRecommendedThreadCount
计算并行训练的推荐线程数 基于CPU核心数和数据批次数量 @param batchCount 数据批次总数 @return 推荐的线程数
src/main/java/io/leavesfly/tinydl/mlearning/parallel/ParallelTrainingUtils.java:73
↓ 2 callersMethodgetStateVariable
将当前状态转换为Variable @return 状态Variable
src/main/java/io/leavesfly/tinydl/modality/rl/environment/CartPoleEnvironment.java:162
↓ 2 callersMethodgetStateVariable
将当前位置转换为状态Variable @return 状态Variable
src/main/java/io/leavesfly/tinydl/modality/rl/environment/GridWorldEnvironment.java:223
↓ 2 callersMethodgetSummary
获取模型的简要信息 @return 简要信息字符串
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:515
↓ 2 callersMethodgetTestDataSet
获取测试数据集 @return 测试数据集
src/main/java/io/leavesfly/tinydl/mlearning/dataset/DataSet.java:85
↓ 2 callersMethodgetTrainingStats
获取训练统计信息 @return 统计信息映射
src/main/java/io/leavesfly/tinydl/modality/rl/agent/DQNAgent.java:378
↓ 2 callersMethodgetUsageRate
获取缓冲区使用率 @return 使用率(0.0 - 1.0)
src/main/java/io/leavesfly/tinydl/modality/rl/ReplayBuffer.java:148
↓ 2 callersMethodgetVocabSize
获取词汇表大小 @return 词汇表大小
src/main/java/io/leavesfly/tinydl/modality/nlp/Word2Vec.java:463
↓ 2 callersMethodgetVocabSize
获取词汇表大小 @return 词汇表大小
src/main/java/io/leavesfly/tinydl/modality/nlp/SimpleTokenizer.java:308
↓ 2 callersMethodgetVocabSize
获取词汇表大小 @return 词汇表大小
src/main/java/io/leavesfly/tinydl/nnet/layer/embedd/Embedding.java:140
↓ 2 callersMethodgetVocabularyTexts
获取用于构建词汇表的文本 @return 用于构建词汇表的文本列表
src/main/java/io/leavesfly/tinydl/example/nlp/GPT2Example.java:91
↓ 2 callersMethodgetXs
获取输入数据数组 @return 输入数据数组
src/main/java/io/leavesfly/tinydl/mlearning/dataset/ArrayDataset.java:122
↓ 2 callersMethodgt
大于比较运算,比较当前数组元素是否大于另一个数组对应元素 @param other 另一个操作数数组 @return 比较结果数组,1.0表示大于,0.0表示不大于 @throws IllegalArgumentException 当两个数组形状不一致时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:587
↓ 2 callersMethodinit
()
src/main/java/io/leavesfly/tinydl/nnet/layer/norm/BatchNorm.java:59
↓ 2 callersMethodinit
()
src/main/java/io/leavesfly/tinydl/nnet/block/MlpBlock.java:57
↓ 2 callersMethodinit
()
src/main/java/io/leavesfly/tinydl/nnet/block/seq2seq/Seq2SeqDecoder.java:62
↓ 2 callersMethodinitState
(NdArray init)
src/main/java/io/leavesfly/tinydl/nnet/block/transformer/TransformerDecoder.java:87
↓ 2 callersMethodinitializeRandomParameters
初始化随机参数
src/test/java/io/leavesfly/tinydl/test/ModelSerializationTest.java:290
↓ 2 callersMethodisModelParallelizable
检查模型是否支持并行训练 主要检查模型是否可序列化 @param model 要检查的模型 @return true 如果支持并行训练
src/main/java/io/leavesfly/tinydl/mlearning/parallel/ParallelTrainingUtils.java:88
↓ 2 callersMethodisSpecialToken
检查是否为特殊token @param token 要检查的token @return 如果是特殊token返回true,否则返回false
src/main/java/io/leavesfly/tinydl/modality/nlp/SimpleTokenizer.java:266
↓ 2 callersMethodlayerForward
前向传播 @param inputs 输入变量数组 @return 前向传播结果
src/main/java/io/leavesfly/tinydl/modality/nlp/GPT2Model.java:221
↓ 2 callersMethodlayerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/MultiHeadAttention.java:75
↓ 2 callersMethodlayerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/PositionalEncoding.java:71
↓ 2 callersMethodlayerForward
(Variable... inputs)
src/main/java/io/leavesfly/tinydl/nnet/layer/transformer/TransformerEncoderLayer.java:67
↓ 2 callersMethodlearn
从经验中学习更新模型 @param experience 经验数据(包含选择的动作和获得的奖励)
src/main/java/io/leavesfly/tinydl/modality/rl/agent/BanditAgent.java:79
↓ 2 callersMethodlearnFromEpisode
回合结束时的学习更新
src/main/java/io/leavesfly/tinydl/modality/rl/agent/REINFORCEAgent.java:249
↓ 2 callersMethodloadModel
加载完整模型 @param filePath 模型文件路径 @param compressed 是否为压缩文件 @return 加载的模型
src/main/java/io/leavesfly/tinydl/mlearning/ModelSerializer.java:78
↓ 2 callersMethodloss
(Variable y, Variable predict)
src/main/java/io/leavesfly/tinydl/mlearning/loss/MeanSquaredLoss.java:15
↓ 2 callersMethodmax
最大值运算 沿指定轴计算变量的最大值 @param _axis 轴索引 @param _keepdims 是否保持维度 @return 最大值运算结果的新变量
src/main/java/io/leavesfly/tinydl/func/Variable.java:496
↓ 2 callersMethodmin
最小值运算 沿指定轴计算变量的最小值 @param _axis 轴索引 @param _keepdims 是否保持维度 @return 最小值运算结果的新变量
src/main/java/io/leavesfly/tinydl/func/Variable.java:510
↓ 2 callersMethodmostSimilar
获取最相似的词 @param word 词 @param topK 返回最相似词的数量 @return 最相似的词列表
src/main/java/io/leavesfly/tinydl/modality/nlp/Word2Vec.java:412
↓ 2 callersMethodnegativeSampling
负采样 @param targetWord 目标词 @param numSamples 采样数量 @return 负样本词列表
src/main/java/io/leavesfly/tinydl/modality/nlp/Word2Vec.java:349
↓ 2 callersMethodplot
绘制模型计算图 通过可视化方式展示模型的前向传播计算过程
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:82
↓ 2 callersMethodpredict
线性预测函数 @param x 输入变量 @param w 权重变量 @param b 偏置变量 @return 预测结果
src/main/java/io/leavesfly/tinydl/example/regress/LineExam.java:91
↓ 2 callersMethodpredictNextToken
预测下一个token
src/main/java/io/leavesfly/tinydl/example/nlp/MoEGPTExample.java:405
↓ 2 callersMethodprintModelInfo
打印模型信息
src/main/java/io/leavesfly/tinydl/modality/nlp/GPT2Model.java:332
↓ 2 callersMethodprintStatistics
打印统计信息 @param algorithmName 算法名称 @param results 结果列表
src/main/java/io/leavesfly/tinydl/example/rl/RLAlgorithmComparison.java:264
↓ 2 callersMethodreset
()
src/main/java/io/leavesfly/tinydl/modality/rl/environment/MultiArmedBanditEnvironment.java:95
↓ 2 callersMethodresetIndex
重置遍历索引
src/main/java/io/leavesfly/tinydl/mlearning/dataset/Batch.java:128
↓ 2 callersMethodresetState
重置解码器状态
src/main/java/io/leavesfly/tinydl/nnet/block/transformer/TransformerDecoder.java:205
↓ 2 callersMethodresumeFromCheckpoint
从检查点恢复模型 @param filePath 检查点文件路径 @return 恢复的模型
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:180
↓ 2 callersMethodrunExperiment
运行实验 @param useBaseline 是否使用基线
src/main/java/io/leavesfly/tinydl/example/rl/GridWorldREINFORCEExample.java:46
↓ 2 callersMethodsampleFromPosterior
从指定臂的后验分布中采样 @param armIndex 臂索引 @return 采样值
src/main/java/io/leavesfly/tinydl/modality/rl/agent/ThompsonSamplingBanditAgent.java:132
↓ 2 callersMethodsaveLogToFile
保存日志到文件 @param logMessage 日志信息
src/main/java/io/leavesfly/tinydl/mlearning/Monitor.java:169
↓ 2 callersMethodsaveModel
保存完整模型(架构 + 参数) @param model 要保存的模型 @param filePath 保存路径 @param compress 是否压缩
src/main/java/io/leavesfly/tinydl/mlearning/ModelSerializer.java:41
↓ 2 callersMethodsaveParameters
仅保存模型参数 @param filePath 保存路径
src/main/java/io/leavesfly/tinydl/mlearning/Model.java:123
↓ 2 callersMethodscalarOperation
通用的与标量运算方法,对数组与标量进行运算 @param scalar 标量值 @param operation 二元运算操作函数 @return 运算结果数组
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:415
↓ 2 callersMethodsetArchitectureType
设置架构类型 @param architectureType 架构类型
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:195
↓ 2 callersMethodsetBestLoss
设置最佳损失值 @param bestLoss 最佳损失值
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:341
↓ 2 callersMethodsetDescription
设置模型描述 @param description 模型描述
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:178
↓ 2 callersMethodsetFinalLoss
设置最终损失值 @param finalLoss 最终损失值
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:324
↓ 2 callersMethodsetModelVersion
设置模型版本 @param modelVersion 模型版本
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:112
↓ 2 callersMethodsetSeed
设置随机种子(用于实验重现) @param seed 随机种子
src/main/java/io/leavesfly/tinydl/modality/rl/agent/EpsilonGreedyBanditAgent.java:156
↓ 2 callersMethodsetShape
设置数组的形状 <p>注意:新形状的大小必须与当前形状大小一致</p> @param shape 新形状 @throws IllegalArgumentException 当新形状大小与当前形状不匹配时抛出
src/main/java/io/leavesfly/tinydl/ndarr/NdArray.java:1590
↓ 2 callersMethodsetTrainedEpochs
设置训练轮次 @param trainedEpochs 训练轮次
src/main/java/io/leavesfly/tinydl/mlearning/ModelInfo.java:307
↓ 2 callersMethodsetX
设置输入数据数组 @param x 输入数据数组
src/main/java/io/leavesfly/tinydl/mlearning/dataset/Batch.java:48
↓ 2 callersMethodsigmoid
Sigmoid运算 对变量执行Sigmoid运算,将值映射到(0,1)区间 @return Sigmoid运算结果的新变量
src/main/java/io/leavesfly/tinydl/func/Variable.java:444
↓ 2 callersMethodsingleThreadTrain
单线程训练(原始实现) @param shuffleData 是否打乱数据
src/main/java/io/leavesfly/tinydl/mlearning/Trainer.java:144
↓ 2 callersMethodsoftMax
SoftMax运算 对变量执行SoftMax运算,常用于多分类问题的输出层 @return SoftMax运算结果的新变量
src/main/java/io/leavesfly/tinydl/func/Variable.java:456
↓ 2 callersMethodstep
(Variable action)
src/main/java/io/leavesfly/tinydl/modality/rl/environment/MultiArmedBanditEnvironment.java:107
↓ 2 callersMethodsubmitGradients
提交一个线程计算的梯度 @param gradients 参数名到梯度的映射
src/main/java/io/leavesfly/tinydl/mlearning/parallel/GradientAggregator.java:43
↓ 2 callersMethodtestAlgorithmAdaptability
测试算法在特定环境下的适应性 @param env 环境 @param envName 环境名称
src/main/java/io/leavesfly/tinydl/example/rl/RLAlgorithmComparison.java:125
← previousnext →301–400 of 1,708, ranked by callers