Class MolEncoder


  • public class MolEncoder
    extends Object
    Format: Each atom is encoded with the same information: 1. One-Hot Atom Information about bonds to all n other atoms. For pair of atom i and j: if i and j are the same, or if i and j are not connected: all zeros otherwise: 1.
    • Constructor Detail

      • MolEncoder

        public MolEncoder()
    • Method Detail

      • setMolecule

        public void setMolecule​(com.actelion.research.chem.StereoMolecule m)
      • encodeMolecule

        public List<BitSet> encodeMolecule()
      • encodeMolecule

        public List<BitSet> encodeMolecule​(int[] permutation)
      • getMoleculeEncodingLength

        public int getMoleculeEncodingLength()
      • getAtomEncodingLength

        public int getAtomEncodingLength()
      • getConnectionEncodingLengthPerConnection

        public int getConnectionEncodingLengthPerConnection()
      • encodeAtomWithConnections

        public BitSet encodeAtomWithConnections​(int zi,
                                                int[] permutation)
      • encodeAtom

        public BitSet encodeAtom​(int zi)
      • encodeConnection

        public BitSet encodeConnection​(int zi,
                                       int zj)
      • oneHot

        public static BitSet oneHot​(int set,
                                    int length)
      • multiHot

        public static BitSet multiHot​(int[] set,
                                      int length)
      • print

        public static void print​(List<BitSet> bs,
                                 int lengthPerBitSet)
      • main

        public static void main​(String[] args)