Skip to content

Latest commit

 

History

History
27 lines (19 loc) · 808 Bytes

README.md

File metadata and controls

27 lines (19 loc) · 808 Bytes
Raw

Decision Trees in Kotlin

This repository contains an implementation of Decision Trees in Kotlin for classification and regression tasks.

Prerequisites

  • Kotlin 1.4 or higher
  • Gradle 4.10 or higher

Getting Started

Clone this repository and build the project using Gradle: gradle build

Usage

import com.example.decisiontree.DecisionTree

// Initialize a decision tree model for binary classification
val decisionTree = DecisionTree(maxDepth = 10)

// Train the model on a training dataset
decisionTree.fit(trainFeatures, trainLabels)

// Predict on a test dataset
val predictions = decisionTree.predict(testFeatures)

##Contributing
We welcome contributions to improve this implementation of Decision Trees in Kotlin. Please submit a pull request if you have any changes to suggest.