DrugPairBatch¶

class DrugPairBatch(identifiers, drug_features_left, drug_molecules_left, drug_features_right, drug_molecules_right, context_features, labels)[source]¶

Bases: object

A data class to store a labeled drug pair batch.

Methods Summary

to(device)

Move this batch to the given device (out of place).

Methods Documentation

to(device)[source]¶

Move this batch to the given device (out of place).

Return type: DrugPairBatch