Struct DatabaseDataset 

pub struct DatabaseDataset { /* private fields */ }

Dataset that loads data from a database table

Implementations

impl DatabaseDataset

pub fn new( connection: Box<dyn DatabaseConnection>, table_name: String, columns: Option<Vec<String>>, batch_size: Option<usize>, ) -> Result<Self, DatabaseError>

Create a new database dataset

pub fn columns(&self) -> &[String]

Get column names

pub fn table_name(&self) -> &str

Get table name

pub fn read_batch( &mut self, start_idx: usize, batch_size: usize, ) -> Result<Vec<DatabaseRow>, DatabaseError>

Read a batch of rows

pub fn rows_to_tensors<T: TensorElement>( &self, rows: &[DatabaseRow], ) -> Result<Vec<Tensor<T>>, DatabaseError>

Convert rows to tensors

Trait Implementations

impl Dataset for DatabaseDataset

type Item = DatabaseRow

The type of items returned by the dataset

fn len(&self) -> usize

Returns the number of items in the dataset

fn get(&self, index: usize) -> Result<Self::Item>

Get a single item from the dataset

fn is_empty(&self) -> bool

Returns true if the dataset is empty