@@ -75,33 +75,13 @@ from polymetrix.featurizers.sidechain_backbone_featurizer import FullPolymerFeat
7575from polymetrix.comparator import PolymerMoleculeComparator
7676
7777# initialize with required featurizers
78+ polymer_featurizer = FullPolymerFeaturizer(MolecularWeight())
79+ molecule_featurizer = FullMolecularFeaturizer(MolecularWeight())
80+
7881polymer = Polymer.from_psmiles(' *CCCCCCNC(=O)c1ccc(C(=O)N*)c(Sc2ccccc2)c1' )
7982molecule = Molecule.from_smiles(' CC(=O)OC1=CC=CC=C1C(=O)O' )
8083
81- polymer_featurizers = [
82- FullPolymerFeaturizer(MolecularWeight()),
83- FullPolymerFeaturizer(NumHBondDonors()),
84- FullPolymerFeaturizer(NumHBondAcceptors()),
85- FullPolymerFeaturizer(NumRotatableBonds())
86- ]
87-
88- molecule_featurizers = [
89- FullMolecularFeaturizer(MolecularWeight()),
90- FullMolecularFeaturizer(NumHBondDonors()),
91- FullMolecularFeaturizer(NumHBondAcceptors()),
92- FullMolecularFeaturizer(NumRotatableBonds())
93- ]
94-
95- polymer_multi = MultipleFeaturizer(polymer_featurizers)
96- molecule_multi = MultipleFeaturizer(molecule_featurizers)
97-
98- comparator = PolymerMoleculeComparator(
99- polymer_multi,
100- molecule_multi,
101- comparisons = [" absolute_difference" , " signed_difference" , " product" , " squared_distance" , " euclidean_distance" ],
102- agg = [" mean" , " max" , " min" , " sum" ]
103- )
104-
84+ comparator = PolymerMoleculeComparator(polymer_featurizer, molecule_featurizer)
10585difference = comparator.compare(polymer, molecule)
10686```
10787
0 commit comments