Commit a0710f95 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

metric

parent 14f0ff9a
Loading
Loading
Loading
Loading
+4 −0
Original line number Diff line number Diff line
Multiclass Examples
-------------------

This directory contains examples of building multiclass models in DeepChem.
+12 −0
Original line number Diff line number Diff line
import deepchem as dc
import numpy as np

N = 10
n_feat = 5
n_classes = 3
n_tasks = 1
X = np.random.rand(N, n_feat)
y = np.random.randint(3, size=(N, n_tasks))
dataset = dc.data.NumpyDataset(X, y)