gcl_classifier / README.md
joesterle's picture
Update README.md
d49b0eb verified
metadata
license: mit
tags:
  - sklearn
  - random-forest
  - calibrated-classifier
library_name: sklearn

Calibrated Random Forest Classifier

This is a calibrated Random Forest classifier trained with scikit-learn on two-photon (2p) ganglion cell layer (GCL) responses from the mouse retina. The training data is from Baden et al. 2016 (https://datadryad.org/dataset/doi:10.5061/dryad.d9v38).

Usage

from gcl_classifier import model

predictions = model.predict(X_test)
probabilities = model.predict_proba(X_test)