Commit 6c224e16 authored by kuehner's avatar kuehner
Browse files

Upload New File

parent b0c0c982
Loading
Loading
Loading
Loading

ontology_create.py

0 → 100644
+65 −0
Original line number Diff line number Diff line
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Sat Mar 13 13:45:20 2021

@author: erkinmenderes
"""
import os
import glob
import os
import subprocess
import sys
import argparse
import re
import csv
import gzip
import igraph
from igraph import *
import matplotlib
from matplotlib import *



def ontology_create(data):
    graph = []
    clean_graph=[]
    n=0
    a=0
    words=[]
    new_tags=[]
    tags=[]
    tag=''
    flag=False
    count=0
    nodes_1=[]
    nodes_2=[]
    vertices=[]
    relation=[]
    g = Graph()
    edges=[]
    with open(data, 'r', encoding ='utf8') as csvfile:
        read = csv.reader(csvfile, delimiter='\t', quotechar='\t') 
        for line in read: 
               if '/c/en/' in line[2] and '/c/en/' in line[3]:
                        nodes_1.append(line[2].replace('/c/en/','').replace('/n','').replace('/v','').replace('/a','').replace('/s','').replace('/r',''))
                        nodes_2.append(line[3].replace('/c/en/','').replace('/n','').replace('/v','').replace('/a','').replace('/s','').replace('/r',''))
                        relation.append(line[1])
                        count=count+1
#                         if count==50:
#                             break
                      
                 
    edges=zip(nodes_1,nodes_2)
    unique_nodes = list(set(nodes_1).union(nodes_2))
    vertex = unique_nodes[:]
    #print(unique_nodes)
    for i in range(len(unique_nodes)):
         g.add_vertices(unique_nodes[i])
    return unique_nodes
onto_list = ontology_create('conceptnet-assertions-5.6.0.csv')

x = onto_list 
y = 0
onto_dict = dict.fromkeys(x,y)