temp fix icews dataset loading

Author: joshloyal

multidynet/datasets/load_icews.py

@@ -17,9 +17,12 @@ def load_icews(dataset='small', country_names='full', year_range=None):
     Y = joblib.load(open(file_name, 'rb'))
 
     if country_names == 'full':
-        countries = np.loadtxt(
-            join(file_path, dir_name, 'numpy_data', 'icews_countries.txt'),
-            delimiter='\n', dtype=np.unicode)
+        #countries = np.loadtxt(
+        #    join(file_path, dir_name, 'numpy_data', 'icews_countries.txt'),
+        #    delimiter='\n', dtype=np.unicode)
+        countries = pd.read_csv(
+            join(file_path, dir_name, 'numpy_data', 'icews_countries.csv')).values.ravel()
+
     else:
         countries = np.loadtxt(
             join(file_path, dir_name, 'numpy_data', 'icews_countries_iso.txt'),

multidynet/datasets/raw_data/icews_large/numpy_data/icews_countries.csv

@@ -0,0 +1,101 @@
+0
+United States
+United Kingdom
+France
+Australia
+China
+India
+Japan
+South Korea
+Russia
+Canada
+South Africa
+Brazil
+Germany
+Iran
+Nigeria
+Spain
+Egypt
+Philippines
+Turkey
+Cuba
+Israel
+Italy
+New Zealand
+Georgia
+Vietnam
+Libya
+Netherlands
+Belgium
+Switzerland
+United Arab Emirates
+Argentina
+Uganda
+Occupied Palestinian Territory
+Thailand
+Kenya
+Qatar
+North Korea
+Pakistan
+Indonesia
+Morocco
+Latvia
+Afghanistan
+Austria
+Mexico
+Singapore
+Taiwan
+Venezuela
+Colombia
+Iraq
+Portugal
+Algeria
+Norway
+Saudi Arabia
+Ethiopia
+Ireland
+Lithuania
+Syria
+Azerbaijan
+Sudan
+Sweden
+Ukraine
+Romania
+Kuwait
+Bulgaria
+Holy See
+Lebanon
+Tunisia
+Greece
+Chile
+Angola
+Czech Republic
+Malaysia
+Poland
+Somalia
+Tanzania
+Bangladesh
+Denmark
+Jordan
+Andorra
+Croatia
+Ghana
+Sri Lanka
+Liberia
+Kazakhstan
+Peru
+Finland
+Senegal
+Haiti
+Estonia
+Serbia
+Slovakia
+Mali
+Yemen
+Zambia
+Zimbabwe
+Rwanda
+Cote d'Ivoire
+Democratic Republic of Congo
+Belarus
+Hungary