MLGodotKit

Empower your Godot projects with the power of machine learning!
MLGodotKit is a C++ GDExtension for Godot, enabling seamless integration of AI-driven features into your games and applications. With support for adaptive behaviors and real-time decision-making, it’s designed to inspire innovation and enhance gameplay.

Node Extensions in Progress

• Data Science Tooling: Tools for preprocessing, analyzing, and visualizing data directly in Godot.
• Deep Learning Integration: Simplified nodes for building and deploying neural networks.
• Cross-Platform Compatibility: Fully supported on Windows, Linux, and macOS.

LIMITATIONS

Please note that this is a work in process pet project of mine and as such will come with a few bugs. If you enjoy the project and want to support please open an issue for desired features or bug fixes!

Next Features to be added:

• A linear layer for the Neural Network Model for regression based predictons
• A robust collection of popular loss functions for out of the box use cases
• More supervised and unsupervised model implementations
• If you'd like to see something else add an issue!

Getting Started

Here are quick examples of how to use the three core models in MLGodotKit:

All models are GDExtension nodes and can be used directly in any Godot scene or script.

Linear Regression (LRNode)

@onready var lr_model = LRNode.new()

func _ready():
    lr_model.set_learning_rate(0.01)
    lr_model.initialize(1)

    # Simple linear dataset: y = 3x + 5
    var inputs = [[1], [2], [3]]
    var targets = [[8], [11], [14]]

    lr_model.train(inputs, targets, 1000)

    var prediction = lr_model.predict([4])
    print("Predicted y for x=4:", prediction)

Descision Tree Classifier (DTreeNode)

@onready var tree = DTreeNode.new()

func _ready():
    tree.set_max_depth(5)
    tree.fit([[1], [2], [3], [10], [11], [12]], [0, 0, 0, 1, 1, 1])

    var result = tree.predict([[2], [11]])
    print("Predictions:", result)  # [0, 1]

Neural Network (NNNode)

Note: All loss functions are left to the user to implement please see examples for suggested implementation will be implemented natively at a later date.

@onready var nn = NNNode.new()

func _ready():
    nn.set_learning_rate(0.1)
    nn.add_layer(2, 4, "relu")
    nn.add_layer(4, 1, "sigmoid")

    var inputs = [[0,0], [0,1], [1,0], [1,1]]
    var targets = [[0], [1], [1], [0]]  # XOR logic

    for i in range(5000):
        for j in range(inputs.size()):
            var y_pred = nn.forward([inputs[j]])[0]
            var error = y_pred - targets[j][0]
            nn.backward([[2.0 * error]])

    print("Test XOR:")
    for i in range(inputs.size()):
        var output = nn.forward([inputs[i]])
        print("Input:", inputs[i], " Predicted:", output, "Expected:", targets[i])

Credits

Built on the powerful Eigen C++ library.