diff options
| -rw-r--r-- | ml_exp/__main__.py | 11 | ||||
| -rw-r--r-- | ml_exp/representations.py | 4 |
2 files changed, 12 insertions, 3 deletions
diff --git a/ml_exp/__main__.py b/ml_exp/__main__.py index 3cb515d85..cb8a9a3f1 100644 --- a/ml_exp/__main__.py +++ b/ml_exp/__main__.py @@ -20,5 +20,14 @@ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. """ +from ml_exp.compound import Compound + + if __name__ == '__main__': - print('Ok!') + print('Initialize test') + test = [] + for i in range(1, 10): + test.append(Compound(f'/home/luevano/py/ml_exp/data/qm7/000{i}.xyz')) + test[i-1].gen_cm(size=1, as_eig=False) + test[i-1].gen_ljm(size=1, as_eig=False) + test[i-1].gen_am(size=1) diff --git a/ml_exp/representations.py b/ml_exp/representations.py index 354611190..ae870f3b0 100644 --- a/ml_exp/representations.py +++ b/ml_exp/representations.py @@ -273,9 +273,9 @@ def adjacency_matrix(fnm, size = n if forces: - am = np.empty((size, size), dtype=float) + am = np.zeros((size, size), dtype=float) else: - am = np.empty((size, size), dtype=int) + am = np.zeros((size, size), dtype=int) for i, bond_i in enumerate(bonds): for j, bond_j in enumerate(bonds): |