Welcome to Battelle’s documentation!

Battelle is a python library designed to create, train and test simple artificial intelligence models.

It was created as a project for the 2021-2022 Artificial Intelligence course of the University of Geneva.

Battelle supports neural networks, perceptrons and KNN classifiers.

Installation

Battelle can be installed with pip:

$ cd battelle
$ pip install .

It is recommended to work in a virtual environment.

Documentation

• Neural Network subpackage (battelle.nn)
  • battelle.nn.Neuron
  • battelle.nn.Layer
  • battelle.nn.NeuralNetwork
  • battelle.nn.ActivationFunction
  • battelle.nn.NeuralNetwork.train
  • battelle.nn.NeuralNetwork.predict
  • battelle.nn.NeuralNetwork.get_metrics
• K-nearest Neighbors Classifier subpackage (battelle.knn)
  • battelle.knn.KNearestNeighborsClassifier
  • battelle.knn.KNearestNeighborsClassifier.predict
  • battelle.knn.KNearestNeighborsClassifier.train
• Perceptron subpackage (battelle.perceptron)
  • battelle.perceptron.Perceptron
  • battelle.perceptron.Perceptron.train
  • battelle.perceptron.Perceptron.predict

Indices and tables

• Index

• Module Index

• Search Page