项目地址

https://github.com/robherley/guesslang-go

guesslang-go 🔍

Go版的yoeo/guesslang。使用深度学习模型检测源代码的编程语言。

设置

依赖项

需要 libtensorflow C API。

在 macOS 上，可以使用 Homebrew 安装：

brew install libtensorflow

或者，在基于 Linux 的系统上：

script/install-libtensorflow

安装

go get github.com/robherley/guesslang-go

在 examples/main.go 中查看示例用法。

注意事项

为了解决Go TensorFlow绑定（和包装库）的一些限制，SavedModel被嵌入到二进制文件中，当初始化一个 Guesser 时，它会将模型临时写入一个目录（然后在之后删除）。

因此，为了使用这个包，您必须至少拥有一个可写的临时目录，与Go的 os.TempDir() 保持一致。

致谢

由以下项目提供支持：

• yoeo/guesslang: 语言模型
• tensorflow/tensorflow: TensorFlow
• galeone/tfgo: 在Go中使用TensorFlow

灵感来源：

• microsoft/vscode-languagedetection
• hieplpvip/guesslang-js

标签：工具分享, 扫描工具, 代码审计